einops datasets tqdm accelerate torchinfo diffusers transformers pathlib safetensors torchvision