loratrainer / requirements.txt
hysts's picture
hysts HF staff
Enable to remove GPU after model training is done
926d3f3
raw
history blame
426 Bytes
accelerate==0.15.0
bitsandbytes==0.36.0.post2
datasets==2.8.0
git+https://github.com/huggingface/diffusers@febaf863026bd014b7a14349336544fc109d0f57#egg=diffusers
ftfy==6.1.1
gradio==3.14.0
git+https://github.com/huggingface/huggingface_hub@bdb9d06b5e67269d702860ca60e1cdb106a66c91#egg=huggingface-hub
Pillow==9.4.0
python-slugify==7.0.0
tensorboard==2.11.2
torch==1.13.1
torchvision==0.14.1
transformers==4.25.1
wandb==0.13.9