From b7c98da30b5a73dbedd48097edf87f001eec79e4 Mon Sep 17 00:00:00 2001
From: PZS-ModelCloud <pzs@modelcloud.ai>
Date: Tue, 19 Nov 2024 09:44:43 +0800
Subject: [PATCH] add gptqmodel[eval] and remove sentencepiece

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

diff --git a/requirements.txt b/requirements.txt
index c769023ff..7266dee3c 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,6 +1,5 @@
 accelerate>=1.1.1
 datasets>=3.1.0
-sentencepiece>=0.2.0
 numpy>=1.26.4
 rouge>=1.0.1
 gekko>=1.1.1
@@ -14,6 +13,5 @@ packaging>=24.2
 ninja>=1.11.1.1
 protobuf>=5.28.3
 huggingface-hub>=0.26.2
-lm_eval==0.4.4
 setuptools>=75.5.0
 device-smi>=0.0.1.dev0
\ No newline at end of file
diff --git a/setup.py b/setup.py
index 05456be1a..d088e4d20 100644
--- a/setup.py
+++ b/setup.py
@@ -249,6 +249,7 @@ def run(self):
         'ipex': ["intel_extension_for_pytorch>=2.5.0"],
         'auto_round': ["auto_round>=0.3"],
         'logger': ["clearml", "random_word", "device-smi", "plotly"],
+        'eval': ["lm_eval>=0.4.4"],
     },
     include_dirs=include_dirs,
     python_requires=">=3.9.0",