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

[nitinmukesh]

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