fix tqdm disable condition

nod-ai · Jul 12, 2024 · 6f0f7c7 · 6f0f7c7
1 parent 6bc8cb4
commit 6f0f7c7
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/models/turbine_models/custom_models/sd_inference/sd_pipeline.py b/models/turbine_models/custom_models/sd_inference/sd_pipeline.py
@@ -565,7 +565,10 @@ def _produce_latents_sdxl(
             [guidance_scale],
             dtype=self.map["unet"]["np_dtype"],
         )
-        for i, t in tqdm(enumerate(timesteps), disable=(self.benchmark and self.verbose)):
+        for i, t in tqdm(
+            enumerate(timesteps),
+            disable=(self.map["unet"].get("benchmark") and self.verbose),
+        ):
             if self.cpu_scheduling:
                 latent_model_input, t = self.scheduler.scale_model_input(
                     latents,