Upload dc.py
Browse files
dc.py
CHANGED
@@ -545,7 +545,7 @@ class GuiSD:
|
|
545 |
info_state = "COMPLETE"
|
546 |
|
547 |
#yield info_state, img, info_images
|
548 |
-
return img,
|
549 |
|
550 |
def dynamic_gpu_duration(func, duration, *args):
|
551 |
|
@@ -685,7 +685,7 @@ def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance
|
|
685 |
gpu_duration = 59
|
686 |
|
687 |
images: list[tuple[PIL.Image.Image, str | None]] = []
|
688 |
-
|
689 |
progress(0, desc="Preparing...")
|
690 |
|
691 |
if randomize_seed:
|
@@ -712,7 +712,7 @@ def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance
|
|
712 |
sd_gen.load_new_model(model_name, vae, TASK_MODEL_LIST[0])
|
713 |
progress(1, desc="Model loaded.")
|
714 |
progress(0, desc="Starting Inference...")
|
715 |
-
images,
|
716 |
guidance_scale, True, generator, lora1, lora1_wt, lora2, lora2_wt, lora3, lora3_wt,
|
717 |
lora4, lora4_wt, lora5, lora5_wt, sampler,
|
718 |
height, width, model_name, vae, TASK_MODEL_LIST[0], None, "Canny", 512, 1024,
|
|
|
545 |
info_state = "COMPLETE"
|
546 |
|
547 |
#yield info_state, img, info_images
|
548 |
+
return info_state, img, info_images
|
549 |
|
550 |
def dynamic_gpu_duration(func, duration, *args):
|
551 |
|
|
|
685 |
gpu_duration = 59
|
686 |
|
687 |
images: list[tuple[PIL.Image.Image, str | None]] = []
|
688 |
+
info_state = info_images = ""
|
689 |
progress(0, desc="Preparing...")
|
690 |
|
691 |
if randomize_seed:
|
|
|
712 |
sd_gen.load_new_model(model_name, vae, TASK_MODEL_LIST[0])
|
713 |
progress(1, desc="Model loaded.")
|
714 |
progress(0, desc="Starting Inference...")
|
715 |
+
info_state, images, info_images = sd_gen_generate_pipeline(prompt, negative_prompt, 1, num_inference_steps,
|
716 |
guidance_scale, True, generator, lora1, lora1_wt, lora2, lora2_wt, lora3, lora3_wt,
|
717 |
lora4, lora4_wt, lora5, lora5_wt, sampler,
|
718 |
height, width, model_name, vae, TASK_MODEL_LIST[0], None, "Canny", 512, 1024,
|