How do I use your model

#5
by VaterLand - opened

I encountered an error when running your code: RuntimeError: The size of tensor a (20) must match the size of tensor b (64) at non-singleton dimension 3. I downloaded all the models together; which one should I use?
fold.png

Ran into the same error. It seems to be an incompability with newer versions of python transformers module. I had to pin it manually to an older version:
pip install transformers==4.46.0

Even when I use transformers==4.46.0, I still get the error: RuntimeError: The size of tensor a (20) must match the size of tensor b (64) at non-singleton dimension 3. If you can run it successfully, could you please send me a copy of the requirements?

I succeeded using transformers==4.44.0 and tokenizers==0.19.1.

Sign up or log in to comment