ford442 commited on
Commit
8701b82
·
verified ·
1 Parent(s): 717b099

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
@@ -109,7 +109,7 @@ class IPAdapter:
109
  clip_image = self.clip_image_processor(images=pil_image, return_tensors="pt").pixel_values
110
  clip_image_embeds = self.image_encoder(clip_image.to(self.device, dtype=torch.bfloat16)).image_embeds
111
  image_prompt_embeds = self.image_proj_model(clip_image_embeds)
112
- print('image_proj_model shape:',image_prompt_embeds)
113
  uncond_image_prompt_embeds = self.image_proj_model(torch.zeros_like(clip_image_embeds))
114
  return image_prompt_embeds, uncond_image_prompt_embeds
115
 
 
109
  clip_image = self.clip_image_processor(images=pil_image, return_tensors="pt").pixel_values
110
  clip_image_embeds = self.image_encoder(clip_image.to(self.device, dtype=torch.bfloat16)).image_embeds
111
  image_prompt_embeds = self.image_proj_model(clip_image_embeds)
112
+ print('image_proj_model shape:',image_prompt_embeds.shape)
113
  uncond_image_prompt_embeds = self.image_proj_model(torch.zeros_like(clip_image_embeds))
114
  return image_prompt_embeds, uncond_image_prompt_embeds
115