wandb
tqdm
transformers
diffusers
safetensors
optimum-quanto
einops