torch==2.1.0
transformers
scikit-learn
numpy<2
sentencepiece