Update app.py

app.py

@@ -24,6 +24,7 @@ controlnet_softEdge_model_path = "lllyasviel/control_v11p_sd15_softedge"
 controlnet_depth_model_path = "lllyasviel/control_v11f1p_sd15_depth"
 # device = "cuda:0"
 device = "cpu"
+dtype = torch.float16 if device.type == 'cuda' else torch.float32
 ##################################################################################################################
 # 1. load pipeline
 ##################################################################################################################
@@ -87,7 +88,7 @@ def task1_StyleTransfer(photo, blur_radius, sketch):
 
     Control_factor = 1.2
     IP_factor = 0.6
-    ip_model = IPAdapter(pipe_depth, image_encoder_path, ip_ckpt, device, Control_factor=Control_factor, IP_factor=IP_factor)
+    ip_model = IPAdapter(pipe_depth, image_encoder_path, ip_ckpt, device, Control_factor=Control_factor, IP_factor=IP_factor，dtype=torch.float32)
 
     depth_image= Image.open(sketch)
     img_array = np.array(depth_image)