Upload app.py

app.py

@@ -148,6 +148,13 @@ img_transform = transforms.Compose([
 
 openpose_model = OpenposeDetector.from_pretrained('lllyasviel/ControlNet').to(args.device)
 
+unet.requires_grad_(False)
+ref_unet.requires_grad_(False)
+image_encoder.requires_grad_(False)
+vae.requires_grad_(False)
+text_encoder.requires_grad_(False)
+
+
 def resize_img(input_image, max_side=640, min_side=512, size=None, 
                pad_to_max_side=False, mode=Image.BILINEAR, base_pixel_number=64):
     w, h = input_image.size
@@ -163,8 +170,7 @@ def resize_img(input_image, max_side=640, min_side=512, size=None,
 @spaces.GPU
 def dress_process(garm_img, face_img, pose_img, prompt, cloth_guidance_scale, caption_guidance_scale,
                   face_guidance_scale,self_guidance_scale, cross_guidance_scale,if_ipa, if_post, if_control, denoise_steps, seed=42):
-    image_face_fusion = pipeline('face_fusion_torch', model='damo/cv_unet_face_fusion_torch',
-                                 model_revision='v1.0.3')
+    image_face_fusion = pipeline('face_fusion_torch', model='damo/cv_unet_face_fusion_torch',)
     if prompt is None:
         prompt = "a photography of a model"
     prompt = prompt + ', best quality, high quality'