requirements.txt

datasets==2.13.1
tokenizers>=0.11.1,!=0.11.3,<0.14
optimum[onnxruntime-gpu]==1.8.8
transformers==4.30.1
accelerate==0.20.3
gradio==3.35.2
--extra-index-url https://download.pytorch.org/whl/cu118
torch==2.0.1