Skip to content

Commit

Permalink
[FIX] check_inputs function in Auraflow Pipeline (#10678)
Browse files Browse the repository at this point in the history
fix_shape_error
  • Loading branch information
SahilCarterr authored Jan 29, 2025
1 parent ea76880 commit aad69ac
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions src/diffusers/pipelines/aura_flow/pipeline_aura_flow.py
Original file line number Diff line number Diff line change
Expand Up @@ -160,8 +160,10 @@ def check_inputs(
prompt_attention_mask=None,
negative_prompt_attention_mask=None,
):
if height % 8 != 0 or width % 8 != 0:
raise ValueError(f"`height` and `width` have to be divisible by 8 but are {height} and {width}.")
if height % (self.vae_scale_factor * 2) != 0 or width % (self.vae_scale_factor * 2) != 0:
raise ValueError(
f"`height` and `width` have to be divisible by {self.vae_scale_factor * 2} but are {height} and {width}."
)

if prompt is not None and prompt_embeds is not None:
raise ValueError(
Expand Down

0 comments on commit aad69ac

Please sign in to comment.