torch>=2.0.0
transformers
pyyaml
