John6666 commited on
Commit
4eb7a16
·
verified ·
1 Parent(s): 9d340bc

Upload dc.py

Browse files
Files changed (1) hide show
  1. dc.py +2 -2
dc.py CHANGED
@@ -373,7 +373,7 @@ class GuiSD:
373
 
374
  def load_new_model(self, model_name, vae_model, task, progress=gr.Progress(track_tqdm=True)):
375
 
376
- yield f"Loading model: {model_name}"
377
 
378
  vae_model = vae_model if vae_model != "None" else None
379
  model_type = get_model_type(model_name)
@@ -393,7 +393,7 @@ class GuiSD:
393
  type_model_precision=dtype_model,
394
  retain_task_model_in_cache=False,
395
  )
396
- yield f"Model loaded: {model_name}"
397
 
398
  @spaces.GPU
399
  @torch.inference_mode()
 
373
 
374
  def load_new_model(self, model_name, vae_model, task, progress=gr.Progress(track_tqdm=True)):
375
 
376
+ #yield f"Loading model: {model_name}"
377
 
378
  vae_model = vae_model if vae_model != "None" else None
379
  model_type = get_model_type(model_name)
 
393
  type_model_precision=dtype_model,
394
  retain_task_model_in_cache=False,
395
  )
396
+ #yield f"Model loaded: {model_name}"
397
 
398
  @spaces.GPU
399
  @torch.inference_mode()