p04798526
/
LLaMA-Factory-Mirror
1
0
Fork 1
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update torch-npu version

This commit is contained in:
MengqingCao 2024-05-29 10:05:11 +00:00
parent e14f5b37e4
commit b19d33b9d2
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"], "qwen": ["tiktoken", "transformers_stream_generator"],
"modelscope": ["modelscope"], "modelscope": ["modelscope"],
"quality": ["ruff"], "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"]
} }