torch>=1.9.0
transformers>=4.0.0
openai
datasets
