accelerate>=0.16.0 torchvision transformers>=4.25.1 ftfy tensorboard Jinja2 diffusers scipy timm fairscale wandb