Update ip_adapter/ip_adapter.py

ip_adapter/ip_adapter.py

@@ -214,7 +214,9 @@ class IPAdapterXL(IPAdapter):
             negative_prompt = [negative_prompt] * num_prompts
         if pil_image_2 != None:
             print('Using secondary image.')
-            image_prompt_embeds, uncond_image_prompt_embeds = self.get_image_embeds(pil_image)
+            image_prompt_embeds_1, uncond_image_prompt_embeds = self.get_image_embeds(pil_image)
+            image_prompt_embeds_2, uncond_image_prompt_embeds_2 = self.get_image_embeds(pil_image_2)
+            image_prompt_embeds = torch.cat(image_prompt_embeds_1,image_prompt_embeds_2).mean(dim=0).unsqueeze(0)  
             bs_embed, seq_len, _ = image_prompt_embeds.shape
             image_prompt_embeds = image_prompt_embeds.repeat(1, num_samples, 1)
             image_prompt_embeds = image_prompt_embeds.view(bs_embed * num_samples, seq_len, -1)