Upload dc.py

dc.py

@@ -677,6 +677,7 @@ def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance
     import numpy as np
     MAX_SEED = np.iinfo(np.int32).max
 
+    image_previews = True
     load_lora_cpu = False
     verbose_info = False
     gpu_duration = 59
@@ -705,8 +706,8 @@ def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance
     progress(1, desc="Preparation completed. Starting inference...")
 
     progress(0, desc="Loading model...")
-    for m in sd_gen.load_new_model(model_name, vae, TASK_MODEL_LIST[0]):
-        print(m)
+    for _ in sd_gen.load_new_model(model_name, vae, TASK_MODEL_LIST[0]):
+        pass
     progress(1, desc="Model loaded.")
     progress(0, desc="Starting Inference...")
     images = None
@@ -716,13 +717,12 @@ def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance
         height, width, model_name, vae, TASK_MODEL_LIST[0], None, "Canny", 512, 1024,
         None, None, None, 0.35, 100, 200, 0.1, 0.1, 1.0, 0., 1., False, "Classic", None,
         1.0, 100, 10, 30, 0.55, "Use same sampler", "", "",
-        False, True, 1, True, False, False, False, False, "./images", False, False, False, True, 1, 0.55,
+        False, True, 1, True, False, image_previews, False, False, "./images", False, False, False, True, 1, 0.55,
         False, False, False, True, False, "Use same sampler", False, "", "", 0.35, True, True, False, 4, 4, 32,
         False, "", "", 0.35, True, True, False, 4, 4, 32,
         True, None, None, "plus_face", "original", 0.7, None, None, "base", "style", 0.7, 0.0,
         load_lora_cpu, verbose_info, gpu_duration
     ):
-        print(info_state)
         images = stream_images
     progress(1, desc="Inference completed.")
     output_image = images[0][0] if images else None