1inkusFace commited on
Commit
2fb476f
Β·
verified Β·
1 Parent(s): 421bd5d

Update ip_adapter/ip_adapter.py

Browse files
Files changed (1) hide show
  1. ip_adapter/ip_adapter.py +1 -1
ip_adapter/ip_adapter.py CHANGED
@@ -270,7 +270,7 @@ class IPAdapterXL(IPAdapter):
270
  image_prompt_embeds_list.append(image_prompt_embeds_5)
271
  uncond_image_prompt_embeds_list.append(uncond_image_prompt_embeds_5)
272
 
273
- image_prompt_embeds = torch.cat(image_prompt_embeds_list).mean(dim=0)
274
  image_prompt_embeds = image_prompt_embeds.mean(dim=0, keepdim=True)
275
  uncond_image_prompt_embeds = torch.cat(uncond_image_prompt_embeds_list, dim=0)
276
  uncond_image_prompt_embeds = uncond_image_prompt_embeds.mean(dim=0, keepdim=True)
 
270
  image_prompt_embeds_list.append(image_prompt_embeds_5)
271
  uncond_image_prompt_embeds_list.append(uncond_image_prompt_embeds_5)
272
 
273
+ image_prompt_embeds = torch.cat(image_prompt_embeds_list)
274
  image_prompt_embeds = image_prompt_embeds.mean(dim=0, keepdim=True)
275
  uncond_image_prompt_embeds = torch.cat(uncond_image_prompt_embeds_list, dim=0)
276
  uncond_image_prompt_embeds = uncond_image_prompt_embeds.mean(dim=0, keepdim=True)