jsonlines==3.1.0
openai==0.27.0
datasets==2.7.0
tenacity==8.1.0
astunparse==1.6.3
torch
transformers
accelerate
xformers