mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add optimum to requirements (for GPTQ LoRA training)

Browse source 
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
Download patch file Download diff file Expand all files Collapse all files

1
requirements.txt
View file

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

1
requirements_nocuda.txt
View file

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