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

Attention error #96

Closed HelloWorldLTY closed 1 day ago

HelloWorldLTY commented 2 weeks ago

Hi, I tried to install the model without triton (which was reported in previous issues), and I received such an error:

assert q.is_cuda and k.is_cuda and v.is_cuda

Would you please share your yml file for the environment to set up DNABERT? I think it will be really helpful. Thanks a lot.

Zhihan1996 commented 1 day ago

please try pip uninstall triton and run the code again.