torch
transformers==4.41.0
sentencepiece==0.1.97
accelerate==0.30.1
bitsandbytes==0.43.1
scikit-learn