AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

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

Browse Source 
fix_shape_error
This commit is contained in:
SahilCarterr
2025-01-30 04:41:54 +05:30
committed by GitHub
parent ea76880bd7
commit aad69ac2f3
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/diffusers/pipelines/aura_flow/pipeline_aura_flow.py
View File

@@ -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(