inference error

#3
by waynezhuwq - opened

I just use the main.py, and it runs error.

File "diffusers/src/diffusers/models/embeddings.py", line 680, in apply_rotary_emb
out = (x.float() * cos + x_rotated.float() * sin).to(x.dtype)
RuntimeError: The size of tensor a (128) must match the size of tensor b (2) at non-singleton dimension 6

alimama-creative org

What versions of transformer and diffusers are you using?

Sign up or log in to comment