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

