Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

AuraFlow pipeline: RuntimeError: shape '[2, 4, 67, 2, 120, 2]' is invalid for input of size 259200 #10673

Closed
nitinmukesh opened this issue Jan 28, 2025 · 0 comments · Fixed by #10678
Labels
bug Something isn't working

Comments

@nitinmukesh
Copy link

Describe the bug

1920x1080 (w x h) throws error

https://github.com/huggingface/diffusers/blob/main/src/diffusers/pipelines/aura_flow/pipeline_aura_flow.py
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}.")

Same as
#10650

Reproduction

import torch
from diffusers import AuraFlowPipeline

pipe = AuraFlowPipeline.from_pretrained("fal/AuraFlow", torch_dtype=torch.float16)
pipe = pipe.to("cuda")
prompt = "A cat holding a sign that says hello world"
image = pipe(prompt, width=1920, height=1080).images[0]
image.save("aura_flow.png")

Logs

former_2d.py", line 89, in forward
    latent = latent.view(
RuntimeError: shape '[2, 4, 67, 2, 120, 2]' is invalid for input of size 259200

System Info

N.A.

Who can help?

No response

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working
Projects
None yet
Development

Successfully merging a pull request may close this issue.

1 participant