torch torchvision numpy tqdm timm pillow diffusers accelerate