torch>=2.0.0
transformers
accelerate
numpy
