Add to requirements

neuralmagic · Jul 18, 2024 · ab3dad3 · ab3dad3
1 parent bbf352f
commit ab3dad3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git 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
@@ -16,6 +16,7 @@
         "datasets",
         "accelerate",
         "tqdm",
+        "llm-compressor @ git+https://github.com/vllm-project/llm-compressor.git"
     ],
     classifiers=[
         "Programming Language :: Python :: 3",