Upload app.py

app.py

@@ -55,7 +55,7 @@ text_encoder = CLIPTextModel.from_pretrained("SG161222/Realistic_Vision_V4.0_noV
 image_encoder = CLIPVisionModelWithProjection.from_pretrained("h94/IP-Adapter", subfolder="models/image_encoder").to(dtype=torch.float16, device=args.device)
 unet = UNet2DConditionModel.from_pretrained("SG161222/Realistic_Vision_V4.0_noVAE", subfolder="unet").to(dtype=torch.float16,device=args.device)
 image_face_fusion = pipeline('face_fusion_torch', model='damo/cv_unet_face_fusion_torch',
-                                     model_revision='v1.0.3', device=args.device)
+                                     model_revision='v1.0.3', device='cpu')
 #face_model
 app = FaceAnalysis(model_path="buffalo_l", providers=[('CUDAExecutionProvider', {"device_id": args.device})]) ##使用GPU:0, 默认使用buffalo_l就可以了
 app.prepare(ctx_id=0, det_size=(640, 640))