requirements.txt

datasets
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
apache-beam==2.48.0
dill==0.3.1.1