From 0576691538da4f93d50b9ee28c252eb91c4f0da3 Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Thu, 31 Aug 2023 08:45:38 -0700
Subject: [PATCH] Add optimum to requirements (for GPTQ LoRA training)

See https://github.com/oobabooga/text-generation-webui/issues/3655
---
 requirements.txt        | 1 +
 requirements_nocuda.txt | 1 +
 2 files changed, 2 insertions(+)

diff --git a/requirements.txt b/requirements.txt
index f6b11cdc..e31d5e7b 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -9,6 +9,7 @@ datasets
 einops
 markdown
 numpy==1.24
+optimum==1.12.0
 pandas
 peft==0.5.*
 Pillow>=9.5.0
diff --git a/requirements_nocuda.txt b/requirements_nocuda.txt
index be0542e0..c560079c 100644
--- a/requirements_nocuda.txt
+++ b/requirements_nocuda.txt
@@ -9,6 +9,7 @@ datasets
 einops
 markdown
 numpy==1.24
+optimum==1.12.0
 pandas
 peft==0.5.*
 Pillow>=9.5.0