Spaces:
Running
on
Zero
Running
on
Zero
Update ip_adapter/ip_adapter.py
Browse files- ip_adapter/ip_adapter.py +1 -2
ip_adapter/ip_adapter.py
CHANGED
@@ -204,7 +204,6 @@ class IPAdapterXL(IPAdapter):
|
|
204 |
pil_image_4=None,
|
205 |
pil_image_5=None,
|
206 |
prompt=None,
|
207 |
-
prompt_2=None,
|
208 |
negative_prompt=None,
|
209 |
text_scale=1.0,
|
210 |
ip_scale=1.0,
|
@@ -284,7 +283,7 @@ class IPAdapterXL(IPAdapter):
|
|
284 |
|
285 |
with torch.inference_mode():
|
286 |
prompt_embeds, negative_prompt_embeds, pooled_prompt_embeds, negative_pooled_prompt_embeds = self.pipe.encode_prompt(
|
287 |
-
prompt
|
288 |
prompt_embeds = prompt_embeds * text_scale
|
289 |
prompt_embeds = torch.cat([prompt_embeds, image_prompt_embeds], dim=1)
|
290 |
negative_prompt_embeds = torch.cat([negative_prompt_embeds, uncond_image_prompt_embeds], dim=1)
|
|
|
204 |
pil_image_4=None,
|
205 |
pil_image_5=None,
|
206 |
prompt=None,
|
|
|
207 |
negative_prompt=None,
|
208 |
text_scale=1.0,
|
209 |
ip_scale=1.0,
|
|
|
283 |
|
284 |
with torch.inference_mode():
|
285 |
prompt_embeds, negative_prompt_embeds, pooled_prompt_embeds, negative_pooled_prompt_embeds = self.pipe.encode_prompt(
|
286 |
+
prompt, num_images_per_prompt=num_samples, do_classifier_free_guidance=True, negative_prompt=negative_prompt)
|
287 |
prompt_embeds = prompt_embeds * text_scale
|
288 |
prompt_embeds = torch.cat([prompt_embeds, image_prompt_embeds], dim=1)
|
289 |
negative_prompt_embeds = torch.cat([negative_prompt_embeds, uncond_image_prompt_embeds], dim=1)
|