requirements.txt

--extra-index-url https://download.pytorch.org/whl/cu118
accelerate==0.20.3
aiohttp==3.8.4
aiosignal==1.3.1
async-timeout==4.0.2
attrs==23.1.0
auto-gptq==0.2.2
bitsandbytes==0.39.0
certifi==2023.5.7
charset-normalizer==3.1.0
cmake==3.26.4
datasets==2.12.0
dill==0.3.6
filelock==3.12.1
frozenlist==1.3.3
fsspec==2023.6.0
huggingface-hub==0.15.1
idna==3.4
Jinja2==3.1.2
jsons==1.6.3
lit==16.0.5.post0
MarkupSafe==2.1.3
mpmath==1.3.0
multidict==6.0.4
multiprocess==0.70.14
networkx==3.1
numpy==1.24.3
packaging==23.1
pandas==2.0.2
protobuf==3.20.3
psutil==5.9.5
pyarrow==12.0.0
python-dateutil==2.8.2
pytz==2023.3
PyYAML==6.0
regex==2023.6.3
requests==2.31.0
responses==0.18.0
rouge==1.0.1
safetensors==0.3.1
scipy==1.10.1
sentencepiece==0.1.99
six==1.16.0
sympy==1.12
tokenizers==0.13.3
torch==2.0.1+cu118
tqdm==4.65.0
transformers==4.30.1
triton==2.0.0
typing_extensions==4.6.3
typish==1.9.3
tzdata==2023.3
urllib3==2.0.3
xxhash==3.2.0
yarl==1.9.2