Skip to content

Commit 91a35e3

Browse files
committed
fix pia test
1 parent 274c33e commit 91a35e3

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

src/diffusers/pipelines/pia/pipeline_pia.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1188,7 +1188,11 @@ def __call__(
11881188
extra_step_kwargs = self.prepare_extra_step_kwargs(generator, eta)
11891189

11901190
# 7. Add image embeds for IP-Adapter
1191-
added_cond_kwargs = {"image_embeds": image_embeds} if ip_adapter_image is not None else None
1191+
added_cond_kwargs = (
1192+
{"image_embeds": image_embeds}
1193+
if ip_adapter_image is not None or ip_adapter_image_embeds is not None
1194+
else None
1195+
)
11921196

11931197
# 8. Denoising loop
11941198
num_warmup_steps = len(timesteps) - num_inference_steps * self.scheduler.order

0 commit comments

Comments
 (0)