[MuseCoCo] models.ARCH_MODEL_REGISTRY[state["args"].arch] about muzic

[XWQlover]

Traceback (most recent call last):
File "interactive_dict_v5_1billion.py", line 33, in
from linear import transformer_lm
File "/home/disk2/xiaoweiqi/music/muzic-main/musecoco/2-attribute2music_model/linear_mask/linear/transformer_lm.py", line 6, in
from .transformer import LinearTransformerDecoder
File "/home/disk2/xiaoweiqi/music/muzic-main/musecoco/2-attribute2music_model/linear_mask/linear/transformer.py", line 23, in
from .transformer_layer import LinearTransformerDecoderLayer
File "/home/disk2/xiaoweiqi/music/muzic-main/musecoco/2-attribute2music_model/linear_mask/linear/transformer_layer.py", line 5, in
from .causal_linear_attention import CausalLinearAttention
File "/home/disk2/xiaoweiqi/music/muzic-main/musecoco/2-attribute2music_model/linear_mask/linear/causal_linear_attention.py", line 12, in
from fast_transformers.causal_product import causal_dot_product
File "/home/disk2/wangjunhao01/anaconda3/envs/MuseCoco/lib/python3.8/site-packages/fast_transformers/causal_product/init.py", line 9, in
from .causal_product_cpu import causal_dot_product as causal_dot_product_cpu,
ImportError: /home/disk2/wangjunhao01/anaconda3/envs/MuseCoco/lib/python3.8/site-packages/fast_transformers/causal_product/causal_product_cpu.cpython-38-x86_64-linux-gnu.so: undefined symbol: _ZN2at4_ops5zeros4callEN3c108ArrayRefINS2_6SymIntEEENS2_8optionalINS2_10ScalarTypeEEENS6_INS2_6LayoutEEENS6_INS2_6DeviceEEENS6_IbEE