wandb
tqdm
transformers
diffusers
safetensors
optimum-quanto
torchvision
sentencepiece
einops
bitsandbytes