fix #4079

2024-06-05 16:56:54 +08:00 · 2024-06-05 16:56:54 +08:00 · 83a005e3d4
parent eef1e542a9
commit 83a005e3d4
2 changed files with 2 additions and 1 deletions
--- a/requirements.txt
+++ b/requirements.txt
@ -7,6 +7,7 @@ gradio>=4.0.0
 scipy
 einops
 sentencepiece
+tiktoken
 protobuf
 uvicorn
 pydantic
--- a/setup.py
+++ b/setup.py
@ -31,7 +31,7 @@ extra_require = {
    "gptq": ["optimum>=1.16.0", "auto-gptq>=0.5.0"],
    "awq": ["autoawq"],
    "aqlm": ["aqlm[gpu]>=1.1.0"],
-    "qwen": ["tiktoken", "transformers_stream_generator"],
+    "qwen": ["transformers_stream_generator"],
    "modelscope": ["modelscope"],
    "quality": ["ruff"],
 }