diff --git a/requirements.txt b/requirements.txt index f40dfeb..3c01461 100644 --- a/requirements.txt +++ b/requirements.txt @@ -3,3 +3,4 @@ transformers datasets accelerate tqdm +llm-compressor @ git+https://github.com/vllm-project/llm-compressor.git diff --git a/setup.py b/setup.py index 7417754..c2b015d 100644 --- a/setup.py +++ b/setup.py @@ -16,6 +16,7 @@ "datasets", "accelerate", "tqdm", + "llm-compressor @ git+https://github.com/vllm-project/llm-compressor.git" ], classifiers=[ "Programming Language :: Python :: 3",