pytest==6.2.5
openai==0.13.0
torch==1.10.2
torchvision==0.11.3
transformers==4.16.2
datasets==1.18.3
sentencepiece==0.1.96
hydra-core==1.1.1
cohere==1.3.1
wandb==0.12.11
nltk==3.7
pytest
matplotlib==3.5.2
omegaconf
protobuf==3.20
