Update ip_adapter/ip_adapter.py

ip_adapter/ip_adapter.py

@@ -107,7 +107,9 @@ class IPAdapter:
         if isinstance(pil_image, Image.Image):
             pil_image = [pil_image]
         clip_image = self.clip_image_processor(images=pil_image, return_tensors="pt").pixel_values
+        print('clip_image_processor shape:',clip_image.shape)
         clip_image_embeds = self.image_encoder(clip_image.to(self.device, dtype=torch.bfloat16)).image_embeds
+        print('image_encoder shape:',clip_image_embeds.shape)
         image_prompt_embeds = self.image_proj_model(clip_image_embeds)
         print('image_proj_model shape:',image_prompt_embeds.shape)
         uncond_image_prompt_embeds = self.image_proj_model(torch.zeros_like(clip_image_embeds))