wandb
tqdm
transformers
diffusers
safetensors
optimum-quanto
torchvision
einops
bitsandbytes