TypeError: dot() got an unexpected keyword argument 'trans_b'

#36
by oagbohouto - opened

Hi,
I try to use this model to get embediings on dna data. I train model on gpu and I get this error. Can someone help me on this.
Regards

I work under google colab and I get A100 GPU

Fond the answer: pip uninstall triton
https://github.com/MAGICS-LAB/DNABERT_2/issues/70

Your need to confirm your account before you can post a new comment.

Sign up or log in to comment