codeGPT / requirements.txt
red1xe's picture
add requirements.txt
d06248f
raw
history blame
158 Bytes
torch==1.13.1 \
torchdata==0.5.1 --quiet
transformers==4.27.2 \
datasets==2.11.0 \
evaluate==0.4.0 \
rouge_score==0.1.2 \
loralib==0.1.1 \
peft==0.3.0 --quiet