transformers
torch
safetensors
einops