diff --git a/vllm_gaudi/v1/worker/hpu_worker.py b/vllm_gaudi/v1/worker/hpu_worker.py index 1d37289af..ae5705e54 100644 --- a/vllm_gaudi/v1/worker/hpu_worker.py +++ b/vllm_gaudi/v1/worker/hpu_worker.py @@ -292,9 +292,8 @@ def initialize_from_config(self, kv_cache_config: KVCacheConfig) -> None: self.compile_or_warm_up_model() def compile_or_warm_up_model(self) -> None: - # Don't run the warmup if in eager or if the model is already warmed up - if not self.model_config.enforce_eager \ - and not getattr(self.model_runner, 'graphed_buckets', None): + # Don't run the warmup if the model is already warmed up + if not getattr(self.model_runner, 'graphed_buckets', None): self.model_runner.warmup_model() # Reset the seed to ensure that the random state is not affected by # the model initialization and profiling.