pillow>=9.4.0
diffusers~=0.30.0
transformers<4.45.0,>=4.42.4
tqdm~=4.66.4
datasets~=2.20.0
bitsandbytes
ftfy
gradio
tensorboard
xformers~=0.0.27

[colab]
torch==2.3.1+cu121
torchvision>=0.18.0
torchaudio==2.3.1+cu121

[cuda]
torch==2.4.0+cu124
torchvision==0.19.0
torchaudio==2.4.0+cu124
