torch
transformers
accelerate
safetensors
