Spaces:
Running
on
Zero
Running
on
Zero
Upload dc.py
Browse files
dc.py
CHANGED
@@ -326,7 +326,7 @@ class GuiSD:
|
|
326 |
img = []
|
327 |
for image in images:
|
328 |
img.append((image, None))
|
329 |
-
return img
|
330 |
|
331 |
def load_new_model(self, model_name, vae_model, task, progress=gr.Progress(track_tqdm=True)):
|
332 |
|
@@ -734,7 +734,7 @@ def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance
|
|
734 |
progress(1, desc="Preparation completed. Starting inference preparation...")
|
735 |
|
736 |
sd_gen.load_new_model(model_name, vae, task_model_list[0], progress)
|
737 |
-
images,
|
738 |
guidance_scale, True, generator, lora1, lora1_wt, lora2, lora2_wt, lora3, lora3_wt,
|
739 |
lora4, lora4_wt, lora5, lora5_wt, sampler,
|
740 |
height, width, model_name, vae, task_model_list[0], None, "Canny", 512, 1024,
|
|
|
326 |
img = []
|
327 |
for image in images:
|
328 |
img.append((image, None))
|
329 |
+
return img
|
330 |
|
331 |
def load_new_model(self, model_name, vae_model, task, progress=gr.Progress(track_tqdm=True)):
|
332 |
|
|
|
734 |
progress(1, desc="Preparation completed. Starting inference preparation...")
|
735 |
|
736 |
sd_gen.load_new_model(model_name, vae, task_model_list[0], progress)
|
737 |
+
images, info = sd_gen.generate_pipeline(prompt, negative_prompt, 1, num_inference_steps,
|
738 |
guidance_scale, True, generator, lora1, lora1_wt, lora2, lora2_wt, lora3, lora3_wt,
|
739 |
lora4, lora4_wt, lora5, lora5_wt, sampler,
|
740 |
height, width, model_name, vae, task_model_list[0], None, "Canny", 512, 1024,
|