diff --git a/setup.py b/setup.py index 5b66a2ea7..432fc5d7e 100644 --- a/setup.py +++ b/setup.py @@ -149,6 +149,9 @@ extras_require = { "vllm": [ "vllm==0.7.2", ], + "llmcompressor": [ + "llm-compressor==0.4.1", + ], } install_requires, dependency_links, extras_require_build = parse_requirements(