423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-14 19:06:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

update torch-npu version

Browse Source 
Former-commit-id: a70d7fcf2967eb30280a1fb845b39db7878f535c
This commit is contained in:
MengqingCao
2024-05-29 10:05:11 +00:00
parent ecada19358
commit c21590ca5e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
setup.py
View File

@@ -33,7 +33,7 @@ extra_require = {
"qwen": ["tiktoken", "transformers_stream_generator"],
"modelscope": ["modelscope"],
"quality": ["ruff"],
"torch_npu": ["torch==2.1.0", "torch_npu==2.1.0", "decorator"]
"torch_npu": ["torch==2.1.0", "torch_npu==2.1.0.post3", "decorator"]
}