Spaces:

haowu11
/

Kolors-Controlnet-Pose-Tryon

Sleeping

haowu11 commited on Feb 6

Commit

98e0801

verified ·

1 Parent(s): aa2a811

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -63,7 +63,7 @@ controlnet = ControlNetModel.from_pretrained( control_path , revision=None).to(d
 face_info_generator = FaceInfoGenerator(root_dir = "./")
-clip_image_encoder = CLIPVisionModelWithProjection.from_pretrained("Kwai-Kolors/Kolors-IP-Adapter-FaceID-Plus",subfolder="clip-vit-large-patch14-336", ignore_mismatched_sizes=True)
 clip_image_encoder.to('cuda')
 clip_image_processor = CLIPImageProcessor(size = 336, crop_size = 336)

 face_info_generator = FaceInfoGenerator(root_dir = "./")
+clip_image_encoder = CLIPVisionModelWithProjection.from_pretrained("Kwai-Kolors/Kolors-IP-Adapter-FaceID-Plus",cache_dir='./',subfolder="clip-vit-large-patch14-336", ignore_mismatched_sizes=True)
 clip_image_encoder.to('cuda')
 clip_image_processor = CLIPImageProcessor(size = 336, crop_size = 336)