Add optimum to requirements (for GPTQ LoRA training)

See https://github.com/oobabooga/text-generation-webui/issues/3655
This commit is contained in:
oobabooga 2023-08-31 08:45:38 -07:00
parent 40ffc3d687
commit 0576691538
2 changed files with 2 additions and 0 deletions

View File

@ -9,6 +9,7 @@ datasets
einops
markdown
numpy==1.24
optimum==1.12.0
pandas
peft==0.5.*
Pillow>=9.5.0

View File

@ -9,6 +9,7 @@ datasets
einops
markdown
numpy==1.24
optimum==1.12.0
pandas
peft==0.5.*
Pillow>=9.5.0