Skip to content

Commit 9fdc07f

Browse files
committed
Change tqdm disable conditions and deepcopy model map on init.
1 parent 2de912e commit 9fdc07f

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

models/turbine_models/custom_models/sd_inference/sd_pipeline.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -247,7 +247,7 @@ def __init__(
247247
"external_weights": None,
248248
"external_weight_path": None,
249249
}
250-
sd_model_map = get_sd_model_map(hf_model_name)
250+
sd_model_map = copy.deepcopy(get_sd_model_map(hf_model_name))
251251
for submodel in sd_model_map:
252252
if "load" not in sd_model_map[submodel]:
253253
sd_model_map[submodel]["load"] = True
@@ -569,9 +569,11 @@ def _produce_latents_sdxl(
569569
[guidance_scale],
570570
dtype=self.map["unet"]["np_dtype"],
571571
)
572+
# Disable progress bar if we aren't in verbose mode or if we're printing
573+
# benchmark latencies for unet.
572574
for i, t in tqdm(
573575
enumerate(timesteps),
574-
disable=(self.map["unet"].get("benchmark") and self.verbose),
576+
disable=(self.map["unet"].get("benchmark") or not self.verbose),
575577
):
576578
if self.cpu_scheduling:
577579
latent_model_input, t = self.scheduler.scale_model_input(

0 commit comments

Comments
 (0)