Update ip_adapter/ip_adapter.py

ip_adapter/ip_adapter.py

@@ -266,10 +266,10 @@ class IPAdapterXL(IPAdapter):
             image_prompt_embeds_list.append(image_prompt_embeds_5)
             uncond_image_prompt_embeds_list.append(uncond_image_prompt_embeds_5)
 
-        #image_prompt_embeds = torch.cat(image_prompt_embeds_list).mean(dim=0).unsqueeze(0)
-        image_prompt_embeds = torch.cat(image_prompt_embeds_list).unsqueeze(0)
+        image_prompt_embeds = torch.cat(image_prompt_embeds_list).mean(dim=0).unsqueeze(0)
+        #image_prompt_embeds = torch.cat(image_prompt_embeds_list).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.repeat(1, num_samples, 1)
         image_prompt_embeds = image_prompt_embeds.view(bs_embed * num_samples, seq_len, -1)
 
         uncond_image_prompt_embeds = torch.cat(uncond_image_prompt_embeds_list).mean(dim=0).unsqueeze(0)