cleanup invalid prints (#18050)

Signed-off-by: calvin chen <120380290@qq.com>
This commit is contained in:
Calvin Chen 2025-05-13 14:01:57 +08:00 committed by GitHub
parent dc1a821768
commit 48545728d8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1554,10 +1554,8 @@ class HPUModelRunnerBase(ModelRunnerBase[TModelInputForHPU]):
'Please update Gaudi Software Suite.')
with compile_only_mode_context(
) if can_use_compile_only_mode else contextlib.nullcontext():
print("aa")
self.warmup_all_buckets(self.bucketing_ctx.prompt_buckets, True,
kv_caches)
print("bb")
self.warmup_all_buckets(self.bucketing_ctx.decode_buckets, False,
kv_caches)