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