add ip-adapter-faceid-plusv2

README.md

@@ -146,10 +146,11 @@ from PIL import Image
 
 from ip_adapter.ip_adapter_faceid import IPAdapterFaceIDPlus
 
+v2 = False
 base_model_path = "SG161222/Realistic_Vision_V4.0_noVAE"
 vae_model_path = "stabilityai/sd-vae-ft-mse"
 image_encoder_path = "h94/IP-Adapter/models/image_encoder"
-ip_ckpt = "ip-adapter-faceid-plus_sd15.bin"
+ip_ckpt = "ip-adapter-faceid-plus_sd15.bin" if not v2 else "ip-adapter-faceid-plusv2_sd15.bin"
 device = "cuda"
 
 noise_scheduler = DDIMScheduler(
@@ -179,7 +180,8 @@ prompt = "photo of a woman in red dress in a garden"
 negative_prompt = "monochrome, lowres, bad anatomy, worst quality, low quality, blurry"
 
 images = ip_model.generate(
-    prompt=prompt, negative_prompt=negative_prompt, face_image=face_image, faceid_embeds=faceid_embeds, num_samples=4, width=512, height=768, num_inference_steps=30, seed=2023
+     prompt=prompt, negative_prompt=negative_prompt, face_image=face_image, faceid_embeds=faceid_embeds, shortcut=v2, s_scale=1.0,
+     num_samples=4, width=512, height=768, num_inference_steps=30, seed=2023
 )
 
 ```