Unverified Commit aad69ac2 authored by SahilCarterr's avatar SahilCarterr Committed by GitHub
Browse files

[FIX] check_inputs function in Auraflow Pipeline (#10678)

fix_shape_error
parent ea76880b
......@@ -160,8 +160,10 @@ class AuraFlowPipeline(DiffusionPipeline):
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(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment