Upload dc.py
Browse files
dc.py
CHANGED
|
@@ -153,7 +153,7 @@ class GuiSD:
|
|
| 153 |
token=True,
|
| 154 |
)
|
| 155 |
|
| 156 |
-
|
| 157 |
if not self.status_loading:
|
| 158 |
self.status_loading = True
|
| 159 |
if i > 0:
|
|
@@ -166,7 +166,7 @@ class GuiSD:
|
|
| 166 |
|
| 167 |
self.status_loading = True
|
| 168 |
|
| 169 |
-
yield f"Loading model: {model_name}"
|
| 170 |
|
| 171 |
if vae_model:
|
| 172 |
vae_type = "SDXL" if "sdxl" in vae_model.lower() else "SD 1.5"
|
|
@@ -217,7 +217,7 @@ class GuiSD:
|
|
| 217 |
self.last_load = datetime.now()
|
| 218 |
self.status_loading = False
|
| 219 |
|
| 220 |
-
|
| 221 |
|
| 222 |
#@spaces.GPU
|
| 223 |
#@torch.inference_mode()
|
|
@@ -327,7 +327,7 @@ class GuiSD:
|
|
| 327 |
pag_scale,
|
| 328 |
):
|
| 329 |
info_state = html_template_message("Navigating latent space...")
|
| 330 |
-
|
| 331 |
|
| 332 |
vae_model = vae_model if vae_model != "None" else None
|
| 333 |
loras_list = [lora1, lora2, lora3, lora4, lora5]
|
|
|
|
| 153 |
token=True,
|
| 154 |
)
|
| 155 |
|
| 156 |
+
for i in range(68):
|
| 157 |
if not self.status_loading:
|
| 158 |
self.status_loading = True
|
| 159 |
if i > 0:
|
|
|
|
| 166 |
|
| 167 |
self.status_loading = True
|
| 168 |
|
| 169 |
+
yield f"Loading model: {model_name}"
|
| 170 |
|
| 171 |
if vae_model:
|
| 172 |
vae_type = "SDXL" if "sdxl" in vae_model.lower() else "SD 1.5"
|
|
|
|
| 217 |
self.last_load = datetime.now()
|
| 218 |
self.status_loading = False
|
| 219 |
|
| 220 |
+
yield f"Model loaded: {model_name}"
|
| 221 |
|
| 222 |
#@spaces.GPU
|
| 223 |
#@torch.inference_mode()
|
|
|
|
| 327 |
pag_scale,
|
| 328 |
):
|
| 329 |
info_state = html_template_message("Navigating latent space...")
|
| 330 |
+
yield info_state, gr.update(), gr.update()
|
| 331 |
|
| 332 |
vae_model = vae_model if vae_model != "None" else None
|
| 333 |
loras_list = [lora1, lora2, lora3, lora4, lora5]
|