search

MAGICS-LAB / DNABERT_2

[ICLR 2024] DNABERT-2: Efficient Foundation Model and Benchmark for Multi-Species Genome
Apache License 2.0
212 stars 49 forks source link

environment about torch version #70

Closed xueleecs closed 4 months ago

xueleecs commented 4 months ago

Thanks for sharing the code. I've tried many versions of torch and still get errors. There's also TypeError("dot() got an unexpected keyword argument 'trans_b'"). Please tell me the possible solution. Thank you so much.

Zhihan1996 commented 4 months ago

Hey,

Can you please try pip uninstall triton? This is the triton package and installation of the package can be very tricky depending on your hardware.