update speed metric condition for number of samples

huggingface · Nov 15, 2024 · 734b15a · 734b15a
1 parent cf1bb31
commit 734b15a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/optimum/habana/diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3.py b/optimum/habana/diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3.py
@@ -618,7 +618,7 @@ def __call__(
                 split=speed_metrics_prefix,
                 start_time=t0,
                 num_samples=num_batches * batch_size
-                if t1 == t0 or num_batches < throughput_warmup_steps
+                if t1 == t0 or use_warmup_inference_steps
                 else (num_batches - throughput_warmup_steps) * batch_size,
                 num_steps=num_batches * batch_size * num_inference_steps,
                 start_time_after_warmup=t1,