Upload dc.py
Browse files
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()
|