From ab3dad3eba0663046a8e291b6984505352c64428 Mon Sep 17 00:00:00 2001
From: mgoin <michael@neuralmagic.com>
Date: Thu, 18 Jul 2024 17:38:02 -0400
Subject: [PATCH] Add to requirements

---
 requirements.txt | 1 +
 setup.py         | 1 +
 2 files changed, 2 insertions(+)

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",