torch
transformers
sentencepiece
accelerate
safetensors