wandb
tqdm
transformers
diffusers
safetensors
torchvision==0.18.1
sentencepiece
einops
bitsandbytes>=0.43.3