Upload dc.py

dc.py

@@ -320,13 +320,13 @@ class GuiSD:
 
     def infer_short(self, model, pipe_params, progress=gr.Progress(track_tqdm=True)):
         progress(0, desc="Start inference...")
-        images, image_list = model(**pipe_params)
+        images, image_list, metadata = model(**pipe_params)
         progress(1, desc="Inference completed.")
         if not isinstance(images, list): images = [images]
         img = []
         for image in images:
             img.append((image, None))
-        return img
+        return img, metadata
 
     def load_new_model(self, model_name, vae_model, task, progress=gr.Progress(track_tqdm=True)):
 
@@ -734,7 +734,7 @@ def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance
     progress(1, desc="Preparation completed. Starting inference preparation...")
 
     sd_gen.load_new_model(model_name, vae, task_model_list[0], progress)
-    images, info = sd_gen.generate_pipeline(prompt, negative_prompt, 1, num_inference_steps,
+    images, metadata, info = sd_gen.generate_pipeline(prompt, negative_prompt, 1, num_inference_steps,
         guidance_scale, True, generator, lora1, lora1_wt, lora2, lora2_wt, lora3, lora3_wt,
         lora4, lora4_wt, lora5, lora5_wt, sampler,
         height, width, model_name, vae, task_model_list[0], None, "Canny", 512, 1024,