requirements.txt

cachetools==5.3.0
certifi==2022.12.7
charset-normalizer==3.1.0
click==8.1.3
cmake==3.26.3
contourpy==1.1.1
cycler==0.12.1
datasets==2.12.0
evaluate==0.4.0
exceptiongroup==1.2.0
filelock==3.12.0
fonttools==4.43.1
frozenlist==1.3.3
fsspec==2024.2.0
GPUtil==1.4.0
httpcore==1.0.2
httpx==0.25.2
huggingface-hub==0.21.1
idna==3.4
importlib-resources==6.1.0
Jinja2==3.1.2
joblib==1.2.0
kiwisolver==1.4.5
lit==16.0.2
MarkupSafe==2.1.2
matplotlib==3.7.3
mpmath==1.3.0
multidict==6.0.4
multiprocess==0.70.14
networkx==3.1
nltk==3.8.1
numpy==1.24.3
nvidia-cublas-cu11==11.10.3.66
nvidia-cuda-cupti-cu11==11.7.101
nvidia-cuda-nvrtc-cu11==11.7.99
nvidia-cuda-runtime-cu11==11.7.99
nvidia-cudnn-cu11==8.5.0.96
nvidia-cufft-cu11==10.9.0.58
nvidia-curand-cu11==10.2.10.91
nvidia-cusolver-cu11==11.4.0.1
nvidia-cusparse-cu11==11.7.4.91
nvidia-ml-py==11.525.112
nvidia-nccl-cu11==2.14.3
nvidia-nvtx-cu11==11.7.91
nvitop==1.1.2
openai==0.28.0
-e git+https://github.com/thunlp/OpenPrompt.git@2aa42a071b1166db4b6c7cc60c580c1f1ed94e5d#egg=openprompt
openpyxl==3.1.2
pandas==2.0.1
peft==0.5.0
Pillow==10.0.1
protobuf==3.20.3
pyarrow==11.0.0
pydantic==2.5.2
pydantic_core==2.14.5
pyparsing==3.1.1
pyphen==0.14.0
pypinyin==0.50.0
python-dateutil
pytz==2023.3
PyYAML==6.0
regex==2023.3.23
requests==2.29.0
responses==0.18.0
rouge==1.0.0
rouge-score==0.1.2
safetensors==0.4.2
scikit-learn==1.2.2
scikit-plot==0.3.7
scipy==1.10.1
sentencepiece==0.1.96
sniffio==1.3.0
sympy==1.11.1
tensorboardX==2.6
termcolor==2.3.0
textstat==0.7.3
threadpoolctl==3.1.0
tokenizers==0.15.2
torch==2.0.1
torchmetrics==0.11.4
torchvision==0.15.2
tqdm==4.65.0
traitlets
transformers==4.38.1
triton==2.0.0
typing_extensions==4.8.0
tzdata==2023.3
urllib3==1.26.15
xlrd==2.0.1
xxhash==3.2.0
yacs==0.1.8
yarl==1.9.2