diff --git a/requirements.txt b/requirements.txt index e17954e3..9e00555e 100644 --- a/requirements.txt +++ b/requirements.txt @@ -7,6 +7,7 @@ gradio>=4.0.0 scipy einops sentencepiece +tiktoken protobuf uvicorn pydantic diff --git a/setup.py b/setup.py index 23f532e7..54f971ca 100644 --- 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"], }