Skip to content

Commit 64cd465

Browse files
authored
Merge branch 'main' into torchao-xpu-int4wo
2 parents 02303d1 + b62614a commit 64cd465

1 file changed

Lines changed: 4 additions & 3 deletions

File tree

src/diffusers/pipelines/z_image/pipeline_z_image_omni.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -588,9 +588,10 @@ def __call__(
588588
negative_prompt_embeds = [npe for npe in negative_prompt_embeds for _ in range(num_images_per_prompt)]
589589

590590
condition_siglip_embeds = [None if sels == [] else sels + [None] for sels in condition_siglip_embeds]
591-
negative_condition_siglip_embeds = [
592-
None if sels == [] else sels + [None] for sels in negative_condition_siglip_embeds
593-
]
591+
if self.do_classifier_free_guidance:
592+
negative_condition_siglip_embeds = [
593+
None if sels == [] else sels + [None] for sels in negative_condition_siglip_embeds
594+
]
594595

595596
actual_batch_size = batch_size * num_images_per_prompt
596597
image_seq_len = (latents.shape[2] // 2) * (latents.shape[3] // 2)

0 commit comments

Comments
 (0)