Update model.py

model.py

@@ -21,9 +21,9 @@ CONTROLNET_MODEL_IDS = {
 }
 
 
-def download_all_controlnet_weights() -> None:
-    for model_id in CONTROLNET_MODEL_IDS.values():
-        ControlNetModel.from_pretrained(model_id)
+# def download_all_controlnet_weights() -> None:
+#     for model_id in CONTROLNET_MODEL_IDS.values():
+#         ControlNetModel.from_pretrained(model_id)
 
 
 class Model:
@@ -57,7 +57,7 @@ class Model:
             beta_schedule="scaled_linear",
             num_train_timesteps=1000,
             steps_offset=1
-        ).to('cuda')
+        )
         # pipe.enable_freeu(b1=1.1, b2=1.1, s1=0.5, s2=0.7)
         pipe.enable_xformers_memory_efficient_attention()
         pipe.force_zeros_for_empty_prompt = False
@@ -176,6 +176,11 @@ class Model:
             prompt=prompt, negative_prompt=negative_prompt, control_image=canny_image, num_steps=num_steps, controlnet_conditioning_scale=float(controlnet_conditioning_scale), seed=seed, num_images=1,
         )
         print(f'Image is {results[0]}')
+        print(prompt)
+        print(negative_prompt)
+        print(num_steps)
+        print(controlnet_conditioning_scale)
+        print(seed)
         return [canny_image, results[0]]