An attempt was made to import the na2d_qk_with'bias function from the natten.functional file, but it did not exist.

[zhangxuan-thecastle]

Dear author, I am very excited after reading your paper and feel that you have written it very well. However, when reproducing the code, I encountered this issue. My natten and torch are both 2.20 and cu118, and there are no other issues. However, the natten.functional function does not include na2d_qk_with'bias

File "/root/data1/zhangxuanxuan/AgileFormer/networks/nat_2d.py", line 30, in from natten.functional import na2d_av, na2d_qk_with_bias ImportError: cannot import name 'na2d_qk_with_bias' from 'natten.functional' (/root/anaconda3/envs/AF/lib/python3.10/site-packages/natten/functional.py)