File tree Expand file tree Collapse file tree 1 file changed +4
-2
lines changed
models/turbine_models/custom_models/sd_inference Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Original file line number Diff line number Diff 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 (
You can’t perform that action at this time.
0 commit comments