requirements.txt

aiohttp==3.9.0
accelerate==0.26.1
aiosignal==1.3.1
appdirs==1.4.4
async-timeout==4.0.2
attrs==23.1.0
beartype==0.14.1
brotlipy==0.7.0
certifi==2023.7.22
cffi==1.15.1
charset-normalizer==3.2.0
click==8.1.6
cmake==3.27.1
contourpy==1.1.0
cycler==0.11.0
datasets==2.14.2
dill==0.3.7
docker-pycreds==0.4.0
einops==0.6.1
exceptiongroup==1.2.0
fancy-einsum==0.0.3
filelock==3.12.2
fonttools==4.25.0
frozenlist==1.4.0
fsspec==2023.6.0
gitdb==4.0.10
GitPython==3.1.41
huggingface-hub==0.16.4
idna==3.4
iniconfig==2.0.0
jaxtyping==0.2.20
Jinja2==3.1.3
kiwisolver==1.4.4
lit==16.0.6
markdown-it-py==3.0.0
MarkupSafe==2.1.3
matplotlib==3.7.2
mdurl==0.1.2
mpmath==1.3.0
multidict==6.0.4
multiprocess==0.70.15
munkres==1.1.4
networkx==3.1
numpy==1.25.2
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-nccl-cu11==2.14.3
nvidia-nvtx-cu11==11.7.91
packaging==23.1
pandas==2.0.3
pathtools==0.1.2
Pillow==10.0.1
pluggy==1.3.0
ply==3.11
protobuf==4.23.4
psutil==5.9.5
pyarrow==14.0.1
pycparser==2.21
Pygments==2.15.1
pyparsing==3.0.9
PyQt5-sip==12.11.0
pytest==7.4.4
python-dateutil==2.8.2
pytz==2023.3
PyYAML==6.0.1
regex==2023.6.3
requests==2.31.0
rich==13.5.1
safetensors==0.3.1
sentencepiece==0.1.99
sentry-sdk==1.29.0
setproctitle==1.3.2
six==1.16.0
smmap==5.0.0
sympy==1.12
tokenizers==0.13.3
tomli==2.0.1
torch==2.0.1
torchaudio==2.0.2
torchvision==0.15.2
tqdm==4.65.0
transformer-lens==1.4.0
transformers==4.36.0
triton==2.0.0
typeguard==3.0.2
typing_extensions==4.7.1
tzdata==2023.3
urllib3==2.0.7
wandb==0.15.7
xxhash==3.3.0
yarl==1.9.2