Update ip_adapter/ip_adapter.py

ip_adapter/ip_adapter.py

@@ -267,7 +267,8 @@ class IPAdapterXL(IPAdapter):
             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
+        #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)