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

fix #4079

This commit is contained in:
hiyouga 2024-06-05 16:56:54 +08:00
parent eef1e542a9
commit 83a005e3d4
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
requirements.txt
View File

@ -7,6 +7,7 @@ gradio>=4.0.0
scipy scipy
einops einops
sentencepiece sentencepiece
tiktoken
protobuf protobuf
uvicorn uvicorn
pydantic pydantic

2
setup.py
View File

@ -31,7 +31,7 @@ extra_require = {
"gptq": ["optimum>=1.16.0", "auto-gptq>=0.5.0"], "gptq": ["optimum>=1.16.0", "auto-gptq>=0.5.0"],
"awq": ["autoawq"], "awq": ["autoawq"],
"aqlm": ["aqlm[gpu]>=1.1.0"], "aqlm": ["aqlm[gpu]>=1.1.0"],
"qwen": ["tiktoken", "transformers_stream_generator"], "qwen": ["transformers_stream_generator"],
"modelscope": ["modelscope"], "modelscope": ["modelscope"],
"quality": ["ruff"], "quality": ["ruff"],
} }