requirements.txt

# for generate (gradio server) and finetune
datasets==2.11.0
sentencepiece==0.1.97
accelerate==0.18.0
gradio==3.27.0
huggingface_hub==0.13.4
appdirs==1.4.4
fire==0.5.0
docutils==0.19
torch==2.0.0
evaluate==0.4.0
rouge_score==0.1.2
sacrebleu==2.3.1
scikit-learn==1.2.2
alt-profanity-check==1.2.2
better-profanity==0.6.1
numpy==1.24.2
pandas==2.0.0
matplotlib==3.7.1
loralib==0.1.1
bitsandbytes==0.38.1
git+https://github.com/huggingface/peft.git@098962fa6515f2e4fe83a757f5995d3ffbb1c373
transformers==4.28.1
tokenizers==0.13.3

# optional for generate
pynvml==11.5.0
psutil==5.9.4
boto3==1.26.101
botocore==1.29.101

# optional for finetune
tensorboard==2.12.1
neptune==1.1.1

# for gradio client
gradio_client==0.1.3
beautifulsoup4==4.12.2
markdown==3.4.1

# data and testing
pytest==7.2.2
pytest-xdist==3.2.1
nltk==3.8.1
textstat==0.7.3
pandoc==2.3
pypandoc==1.11
openpyxl==3.1.2
lm_dataformat==0.0.20
bioc==2.0