requirements.txt

absl-py==1.4.0
accelerate==0.12.0
aiohttp==3.8.4
aiosignal==1.3.1
appdirs==1.4.4
async-timeout==4.0.2
attrs==22.2.0
blobfile==2.0.1
certifi==2022.12.7
charset-normalizer==3.0.1
click==8.1.3
cmake==3.26.0
contextlib2==21.6.0
datasets==2.10.1
dill==0.3.6
docker-pycreds==0.4.0
einops==0.6.0
filelock==3.9.0
frozenlist==1.3.3
fsspec==2023.1.0
ftfy==6.1.1
gitdb==4.0.10
GitPython==3.1.31
huggingface-hub==0.12.1
idna==3.4
lit==16.0.0
lxml==4.9.2
ml-collections==0.1.1
multidict==6.0.4
multiprocess==0.70.14
mypy-extensions==1.0.0
numpy==1.24.2
nvidia-cublas-cu11==11.10.3.66
nvidia-cuda-nvrtc-cu11==11.7.99
nvidia-cuda-runtime-cu11==11.7.99
nvidia-cudnn-cu11==8.5.0.96
packaging==23.0
pandas==1.5.3
pathtools==0.1.2
Pillow==9.4.0
pip==22.3.1
protobuf==4.22.0
psutil==5.9.4
pyarrow==11.0.0
pycryptodomex==3.17
pyre-extensions==0.0.23
PySocks==1.7.1
python-dateutil==2.8.2
pytz==2022.7.1
PyYAML==6.0
regex==2022.10.31
requests==2.28.2
responses==0.18.0
sentry-sdk==1.16.0
setproctitle==1.3.2
setuptools==65.6.3
six==1.16.0
smmap==5.0.0
tokenizers==0.13.2
torch==1.13.1+cu116
torchaudio==0.13.1+cu116
torchvision==0.14.1+cu116
tqdm==4.64.1
transformers==4.23.1
triton==2.0.0.post1
typing_extensions==4.5.0
typing-inspect==0.8.0
urllib3==1.26.14
wandb==0.13.10
wcwidth==0.2.6
wheel==0.38.4
xformers==0.0.16
xxhash==3.2.0
yarl==1.8.2