Add optimum to requirements (for GPTQ LoRA training)

See #3655
oobabooga · Aug 31, 2023 · 0576691 · 0576691
1 parent 40ffc3d
commit 0576691
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git 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
@@ -9,6 +9,7 @@ datasets
 einops
 markdown
 numpy==1.24
+optimum==1.12.0
 pandas
 peft==0.5.*
 Pillow>=9.5.0