File tree 1 file changed +4
-2
lines changed
models/turbine_models/custom_models/sd_inference
1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -247,7 +247,7 @@ def __init__(
247
247
"external_weights" : None ,
248
248
"external_weight_path" : None ,
249
249
}
250
- sd_model_map = get_sd_model_map (hf_model_name )
250
+ sd_model_map = copy . deepcopy ( get_sd_model_map (hf_model_name ) )
251
251
for submodel in sd_model_map :
252
252
if "load" not in sd_model_map [submodel ]:
253
253
sd_model_map [submodel ]["load" ] = True
@@ -569,9 +569,11 @@ def _produce_latents_sdxl(
569
569
[guidance_scale ],
570
570
dtype = self .map ["unet" ]["np_dtype" ],
571
571
)
572
+ # Disable progress bar if we aren't in verbose mode or if we're printing
573
+ # benchmark latencies for unet.
572
574
for i , t in tqdm (
573
575
enumerate (timesteps ),
574
- disable = (self .map ["unet" ].get ("benchmark" ) and self .verbose ),
576
+ disable = (self .map ["unet" ].get ("benchmark" ) or not self .verbose ),
575
577
):
576
578
if self .cpu_scheduling :
577
579
latent_model_input , t = self .scheduler .scale_model_input (
You can’t perform that action at this time.
0 commit comments