torch>=1.6.0
transformers==4.34.0
datasets
accelerate>=0.20.1
sentence_transformers
