Skip to content

GitLab

Unverified Commit 48545728 authored by Calvin Chen's avatar Calvin Chen Committed by GitHub
Browse files

cleanup invalid prints (#18050) 


Signed-off-by: default avatarcalvin chen <120380290@qq.com>
parent dc1a8217
Show whitespace changes
Inline Side-by-side
Showing with 0 additions and 2 deletions
vllm/worker/hpu_model_runner.py
View file @ 48545728
...@@ -1554,10 +1554,8 @@ class HPUModelRunnerBase(ModelRunnerBase[TModelInputForHPU]): ...@@ -1554,10 +1554,8 @@ class HPUModelRunnerBase(ModelRunnerBase[TModelInputForHPU]):
'Please update Gaudi Software Suite.') 'Please update Gaudi Software Suite.')
with compile_only_mode_context( with compile_only_mode_context(
) if can_use_compile_only_mode else contextlib.nullcontext(): ) if can_use_compile_only_mode else contextlib.nullcontext():
print("aa")
self.warmup_all_buckets(self.bucketing_ctx.prompt_buckets, True, self.warmup_all_buckets(self.bucketing_ctx.prompt_buckets, True,
kv_caches) kv_caches)
print("bb")
self.warmup_all_buckets(self.bucketing_ctx.decode_buckets, False, self.warmup_all_buckets(self.bucketing_ctx.decode_buckets, False,
kv_caches) kv_caches)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment