Upload dc.py

dc.py

@@ -747,11 +747,11 @@ def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance
     lora5 = get_valid_lora_path(lora5)
     progress(1, desc="Preparation completed. Starting inference preparation...")
 
-    sd_gen.load_new_model(model_name, vae, task_model_list[0], progress)
+    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,
         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,
+        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,