Update ip_adapter/ip_adapter.py

ip_adapter/ip_adapter.py

@@ -125,10 +125,6 @@ class IPAdapter:
         self,
         pil_image,
         prompt=None,
-        prompt2=None,
-        prompt3=None,
-        prompt4=None,
-        prompt5=None,
         negative_prompt=None,
         scale=1.0,
         num_samples=4,
@@ -173,22 +169,6 @@ class IPAdapter:
                 prompt, device=self.device, num_images_per_prompt=num_samples, do_classifier_free_guidance=True, negative_prompt=negative_prompt)
             negative_prompt_embeds_, prompt_embeds_ = prompt_embeds.chunk(2)
             prompt_embeds_list.append(prompt_embeds)
-            if prompt2 is not None:
-                prompt_embeds = self.pipe._encode_prompt(
-                    prompt2, device=self.device, num_images_per_prompt=num_samples, do_classifier_free_guidance=True, negative_prompt=negative_prompt)
-            prompt_embeds_list.append(prompt_embeds)
-            if prompt3 is not None:
-                prompt_embeds = self.pipe._encode_prompt(
-                    prompt3, device=self.device, num_images_per_prompt=num_samples, do_classifier_free_guidance=True, negative_prompt=negative_prompt)
-            prompt_embeds_list.append(prompt_embeds)
-            if prompt4 is not None:
-                prompt_embeds = self.pipe._encode_prompt(
-                    prompt4, device=self.device, num_images_per_prompt=num_samples, do_classifier_free_guidance=True, negative_prompt=negative_prompt)
-            prompt_embeds_list.append(prompt_embeds)
-            if prompt5 is not None:
-                prompt_embeds = self.pipe._encode_prompt(
-                    prompt5, device=self.device, num_images_per_prompt=num_samples, do_classifier_free_guidance=True, negative_prompt=negative_prompt)
-            prompt_embeds_list.append(prompt_embeds)
             
             prompt_embeds = torch.cat([prompt_embeds_, image_prompt_embeds], dim=1)
             negative_prompt_embeds = torch.cat([negative_prompt_embeds_, uncond_image_prompt_embeds], dim=1)
@@ -226,6 +206,7 @@ class IPAdapterXL(IPAdapter):
         pil_image_4=None,
         pil_image_5=None,
         prompt=None,
+        prompt1=None,
         prompt2=None,
         prompt3=None,
         prompt4=None,
@@ -315,6 +296,11 @@ class IPAdapterXL(IPAdapter):
                 prompt, num_images_per_prompt=num_samples, do_classifier_free_guidance=True, negative_prompt=negative_prompt)
             prompt_embeds_list.append(prompt_embeds)
             pooled_prompt_embeds_list.append(pooled_prompt_embeds)
+            if prompt1 is not None:
+                prompt_embeds, negative_prompt_embeds_, pooled_prompt_embeds, negative_pooled_prompt_embeds_ = self.pipe.encode_prompt(
+                    prompt1, num_images_per_prompt=num_samples, do_classifier_free_guidance=True, negative_prompt=negative_prompt)
+                prompt_embeds_list.append(prompt_embeds)
+                pooled_prompt_embeds_list.append(pooled_prompt_embeds)
             if prompt2 is not None:
                 prompt_embeds, negative_prompt_embeds_, pooled_prompt_embeds, negative_pooled_prompt_embeds_ = self.pipe.encode_prompt(
                     prompt2, num_images_per_prompt=num_samples, do_classifier_free_guidance=True, negative_prompt=negative_prompt)