"tests/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "cf4227cd1e1a361aaf26109f2e970aa9abb620b7"
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): ...@@ -160,8 +160,10 @@ class AuraFlowPipeline(DiffusionPipeline):
prompt_attention_mask=None, prompt_attention_mask=None,
negative_prompt_attention_mask=None, negative_prompt_attention_mask=None,
): ):
if height % 8 != 0 or width % 8 != 0: 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 8 but are {height} and {width}.") 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: if prompt is not None and prompt_embeds is not None:
raise ValueError( 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