Update ip_adapter/ip_adapter.py

ip_adapter/ip_adapter.py

@@ -266,11 +266,11 @@ 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)
-        bs_embed, seq_len = image_prompt_embeds.shape
-        #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)
+        #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.view(bs_embed * num_samples, seq_len, -1)
 
         uncond_image_prompt_embeds = torch.cat(uncond_image_prompt_embeds_list).mean(dim=0).unsqueeze(0)  
         uncond_image_prompt_embeds = uncond_image_prompt_embeds.repeat(1, num_samples, 1)