torch
transformers
torchvision
einops
safetensors