Skip to content

Commit aad69ac

Browse files
authored
[FIX] check_inputs function in Auraflow Pipeline (#10678)
fix_shape_error
1 parent ea76880 commit aad69ac

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

src/diffusers/pipelines/aura_flow/pipeline_aura_flow.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -160,8 +160,10 @@ def check_inputs(
160160
prompt_attention_mask=None,
161161
negative_prompt_attention_mask=None,
162162
):
163-
if height % 8 != 0 or width % 8 != 0:
164-
raise ValueError(f"`height` and `width` have to be divisible by 8 but are {height} and {width}.")
163+
if height % (self.vae_scale_factor * 2) != 0 or width % (self.vae_scale_factor * 2) != 0:
164+
raise ValueError(
165+
f"`height` and `width` have to be divisible by {self.vae_scale_factor * 2} but are {height} and {width}."
166+
)
165167

166168
if prompt is not None and prompt_embeds is not None:
167169
raise ValueError(

0 commit comments

Comments
 (0)