[diffusion] fix: avoid saving output for warmup requests (#18867)

This commit is contained in:
Mick
2026-02-16 11:48:28 +08:00
committed by GitHub
parent 8a82c70297
commit 78b4c9e248
2 changed files with 3 additions and 2 deletions

View File

@@ -198,7 +198,6 @@ class Scheduler:
prompt="",
negative_prompt="",
image_path=[input_path],
is_warmup=True,
)
else:
req = Req(
@@ -206,8 +205,8 @@ class Scheduler:
width=width,
height=height,
prompt="",
is_warmup=True,
)
req.set_as_warmup()
self.waiting_queue.append((None, req))
# if server is warmed-up, set this flag to avoid req-based warmup
self.warmed_up = True

View File

@@ -252,6 +252,8 @@ class Req:
def set_as_warmup(self):
self.is_warmup = True
self.save_output = False
self.suppress_logs = True
self.extra["cache_dit_num_inference_steps"] = self.num_inference_steps
self.num_inference_steps = 1