KakaruHayate f7d3d32cf0 在其他推理部分加载torch_musa

2024-03-30 22:14:45 +08:00

..

__init__.py

Add files via upload

2024-01-16 17:38:48 +08:00

activation_onnx.py

Add AR Onnx Module

2024-01-25 02:31:08 +08:00

activation.py

在其他推理部分加载torch_musa

2024-03-30 22:14:45 +08:00

embedding_onnx.py

Add AR Onnx Module

2024-01-25 02:31:08 +08:00

embedding.py

在其他推理部分加载torch_musa

2024-03-30 22:14:45 +08:00

lr_schedulers.py

修改代码引用，淡定

2024-02-28 17:31:19 +08:00

optim.py

Add files via upload

2024-01-16 17:38:48 +08:00

patched_mha_with_cache_onnx.py

Add AR Onnx Module

2024-01-25 02:31:08 +08:00

patched_mha_with_cache.py

在其他推理部分加载torch_musa

2024-03-30 22:14:45 +08:00

scaling.py

在其他推理部分加载torch_musa

2024-03-30 22:14:45 +08:00

transformer_onnx.py

Add AR Onnx Module

2024-01-25 02:31:08 +08:00

transformer.py

在其他推理部分加载torch_musa

2024-03-30 22:14:45 +08:00