John6666 commited on
Commit
86daee1
1 Parent(s): 77058ee

Upload dc.py

Browse files
Files changed (1) hide show
  1. dc.py +2 -2
dc.py CHANGED
@@ -326,7 +326,7 @@ class GuiSD:
326
  img = []
327
  for image in images:
328
  img.append((image, None))
329
- return img, metadata
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, metadata, 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,
 
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,