Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 2 additions & 3 deletions vllm_gaudi/v1/worker/hpu_worker.py
Original file line number Diff line number Diff line change
Expand Up @@ -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.
Expand Down
Loading