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]
|