Update app.py

app.py

@@ -20,11 +20,12 @@ high_threshold = 200
 # Models
 pose_model = OpenposeDetector.from_pretrained("lllyasviel/ControlNet")
 controlnet = ControlNetModel.from_pretrained(
-    "lllyasviel/sd-controlnet-openpose", torch_dtype=torch.float16
+    "lllyasviel/sd-controlnet-openpose"
 )
 pipe = StableDiffusionControlNetPipeline.from_pretrained(
-    "runwayml/stable-diffusion-v1-5", controlnet=controlnet, safety_checker=None, torch_dtype=torch.float16
+    "runwayml/stable-diffusion-v1-5", controlnet=controlnet, safety_checker=None
 )
+pipe = pipe.to("cpu")
 
 
 def get_pose(image):
@@ -87,8 +88,7 @@ def inference(prompt,canny_image,number,seed ):
     # need to explicitly call pipe.to("cuda").
     #pipe.enable_model_cpu_offload()
     # xformers
-    pipe.enable_xformers_memory_efficient_attention()
-    torch.cuda.is_available() 
+    #pipe.enable_xformers_memory_efficient_attention()
     # Generator seed,
     generator = torch.manual_seed(seed)
     image_ = canny_image