John6666 commited on
Commit
0c8469e
·
verified ·
1 Parent(s): 91577c9

Upload dc.py

Browse files
Files changed (1) hide show
  1. dc.py +3 -3
dc.py CHANGED
@@ -545,7 +545,7 @@ class GuiSD:
545
  info_state = "COMPLETE"
546
 
547
  #yield info_state, img, info_images
548
- return img, info_state
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
- info: str = ""
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, info = 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,
 
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,