Shape error

#9
by Abdrah - opened

I Tried the snippet on the model card and got: RuntimeError: shape '[1, 59, 64, 128]' is invalid for input of size 60416.
I used a DGX station with 4 A100 80BG.

fixed

just update transformers library

Thanks, it is working now. I was using the version 4.30.2, and now I'm using 4.31.0 and it is working fine.

Abdrah changed discussion status to closed

Sign up or log in to comment