diff --git a/setup.py b/setup.py index 0f3892c3b..239a00edb 100644 --- a/setup.py +++ b/setup.py @@ -149,6 +149,9 @@ extras_require = { "vllm": [ "vllm==0.7.2", ], + "llmcompressor": [ + "llmcompressor~=0.5.0", + ], } install_requires, dependency_links, extras_require_build = parse_requirements(