transformers
torch
safetensors
accelerate