Update ip_adapter/ip_adapter.py

ip_adapter/ip_adapter.py

@@ -272,7 +272,8 @@ class IPAdapterXL(IPAdapter):
 
         image_prompt_embeds = torch.cat(image_prompt_embeds_list).mean(dim=0)
         image_prompt_embeds = image_prompt_embeds.mean(dim=0, keepdim=True)
-        uncond_image_prompt_embeds = torch.cat(uncond_image_prompt_embeds_list).mean(dim=0)
+        uncond_image_prompt_embeds = torch.cat(uncond_image_prompt_embeds_list, dim=0)
+        uncond_image_prompt_embeds = uncond_image_prompt_embeds.mean(dim=0, keepdim=True)
 
         with torch.inference_mode():
             prompt_embeds, negative_prompt_embeds, pooled_prompt_embeds, negative_pooled_prompt_embeds = self.pipe.encode_prompt(