Update app.py

app.py

@@ -35,17 +35,26 @@ high_threshold = 200
 
 # Models
 pose_model = OpenposeDetector.from_pretrained("lllyasviel/ControlNet")
+'''
 controlnet = ControlNetModel.from_pretrained(
     "lllyasviel/sd-controlnet-openpose", torch_dtype=torch.float16
 )
 pipe = StableDiffusionControlNetPipeline.from_pretrained(
     "runwayml/stable-diffusion-v1-5", controlnet=controlnet, safety_checker=None, torch_dtype=torch.float16
 )
+'''
+
+controlnet = ControlNetModel.from_pretrained(
+    "lllyasviel/sd-controlnet-openpose", torch_dtype=torch.get_default_dtype()
+)
+pipe = StableDiffusionControlNetPipeline.from_pretrained(
+    "runwayml/stable-diffusion-v1-5", controlnet=controlnet, safety_checker=None, torch_dtype=torch.get_default_dtype()
+)
 pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
 
 # This command loads the individual model components on GPU on-demand. So, we don't
 # need to explicitly call pipe.to("cuda").
-pipe.enable_model_cpu_offload()
+#pipe.enable_model_cpu_offload()
 
 # xformers
 pipe.enable_xformers_memory_efficient_attention()