Don't redundantly use tqdm progress if we're printing benchmarks

nod-ai · Jul 12, 2024 · 6bc8cb4 · 6bc8cb4
1 parent a7de2ba
commit 6bc8cb4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/models/turbine_models/custom_models/pipeline_base.py b/models/turbine_models/custom_models/pipeline_base.py
@@ -343,7 +343,8 @@ def __init__(
         common_export_args: dict = {},
     ):
         self.map = model_map
-        self.printer = Printer(verbose, time.time(), True)
+        self.verbose = verbose
+        self.printer = Printer(self.verbose, time.time(), True)
         if isinstance(device, dict):
             assert isinstance(
                 target, dict

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,7 @@ def _produce_latents_sdxl(
             [guidance_scale],
             dtype=self.map["unet"]["np_dtype"],
         )
-        for i, t in tqdm(enumerate(timesteps)):
+        for i, t in tqdm(enumerate(timesteps), disable=(self.benchmark and self.verbose)):
             if self.cpu_scheduling:
                 latent_model_input, t = self.scheduler.scale_model_input(
                     latents,