Update app.py

app.py

@@ -66,7 +66,7 @@ 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_processor = CLIPImageProcessor(size = 336, crop_size = 336)
-print(os.listdir('./models--Kwai-Kolors--Kolors-IP-Adapter-FaceID-Plus'))
+snapshotname = os.listdir('./models--Kwai-Kolors--Kolors-IP-Adapter-FaceID-Plus/snapshots')[0]
 pipe = StableDiffusionXLControlNetImg2ImgPipeline(
         vae=vae,
         controlnet = controlnet,
@@ -83,7 +83,7 @@ pipe = StableDiffusionXLControlNetImg2ImgPipeline(
 if hasattr(pipe.unet, 'encoder_hid_proj'):
     pipe.unet.text_encoder_hid_proj = pipe.unet.encoder_hid_proj
 ip_scale = 0.5
-pipe.load_ip_adapter_faceid_plus('models--Kwai-Kolors--Kolors-IP-Adapter-FaceID-Plus/ipa-faceid-plus.bin', device = 'cuda')
+pipe.load_ip_adapter_faceid_plus(f'models--Kwai-Kolors--Kolors-IP-Adapter-FaceID-Plus/snapshots/{snapshotname}/ipa-faceid-plus.bin', device = 'cuda')
 pipe.set_face_fidelity_scale(ip_scale)
 pipe = pipe.to("cuda")
 pipe.enable_model_cpu_offload()