requirements.txt

--extra-index-url https://download.pytorch.org/whl/cu116
omegaconf==2.3.0
peft==0.5.0
pyyaml==6.0.1
sentencepiece
torch==1.13.0+cu116
transformers==4.28.1