423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-08-22 05:32:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

update torch-npu version

Browse Source 
Former-commit-id: b19d33b9d205911ffd191bd0eadddd162b692e68
This commit is contained in:
MengqingCao 2024-05-29 10:05:11 +00:00
parent 29fe1cd688
commit 2192616770
1 changed files with 1 additions and 1 deletions
Show Stats 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"]
}