Update ip_adapter/ip_adapter.py

ip_adapter/ip_adapter.py

@@ -25,7 +25,7 @@ class ImageProjModel(torch.nn.Module):
         clip_extra_context_tokens = self.proj(embeds).reshape(-1, self.clip_extra_context_tokens, self.cross_attention_dim)
         clip_extra_context_tokens = self.norm(clip_extra_context_tokens)
         return clip_extra_context_tokens
-
+'''
 class IPAdapter:
     
     def __init__(self, sd_pipe, image_encoder_path, ip_ckpt, device, num_tokens=4):
@@ -180,7 +180,7 @@ class IPAdapter:
         
         return images
     
-    
+'''
 class IPAdapterXL(IPAdapter):
     """SDXL"""
     
@@ -236,7 +236,8 @@ class IPAdapterXL(IPAdapter):
             
         image_prompt_embeds_list = []
         uncond_image_prompt_embeds_list = []
-        
+        print('Using primary image.')
+
         image_prompt_embeds_1, uncond_image_prompt_embeds_1 = self.get_image_embeds(pil_image_1)
         image_prompt_embeds_list.append(image_prompt_embeds_1)
         uncond_image_prompt_embeds_list.append(uncond_image_prompt_embeds_1)
@@ -254,13 +255,13 @@ class IPAdapterXL(IPAdapter):
             image_prompt_embeds_list.append(image_prompt_embeds_3)
             uncond_image_prompt_embeds_list.append(uncond_image_prompt_embeds_3)
         if pil_image_4 != None:
-            print('Using secondary image.')
+            print('Using quaternary image.')
             image_prompt_embeds_4, uncond_image_prompt_embeds_4 = self.get_image_embeds(pil_image_4)
             image_prompt_embeds_4 = image_prompt_embeds_4 * scale_4
             image_prompt_embeds_list.append(image_prompt_embeds_4)
             uncond_image_prompt_embeds_list.append(uncond_image_prompt_embeds_4)
         if pil_image_5 != None:
-            print('Using secondary image.')
+            print('Using quinary image.')
             image_prompt_embeds_5, uncond_image_prompt_embeds_5 = self.get_image_embeds(pil_image_5)
             image_prompt_embeds_5 = image_prompt_embeds_5 * scale_5
             image_prompt_embeds_list.append(image_prompt_embeds_5)