diff --git a/setup.py b/setup.py index 15ea648a..7af9b825 100644 --- a/setup.py +++ b/setup.py @@ -54,7 +54,7 @@ extra_require = { "awq": ["autoawq"], "aqlm": ["aqlm[gpu]>=1.1.0"], "vllm": ["vllm>=0.4.3,<=0.7.3"], - "sglang": ["sglang>=0.4.4"], + "sglang": ["sglang[srt]>=0.4.4", "transformers==4.48.3"], "galore": ["galore-torch"], "apollo": ["apollo-torch"], "badam": ["badam>=1.2.1"],