requirements.txt

aiohttp>=3.8.5
aiosignal>=1.3.1
appdirs>=1.4.4
async-timeout>=4.0.3
attrs>=23.1.0
axial-positional-embedding>=0.2.1
cachetools>=5.3.1
certifi>=2023.7.22
charset-normalizer>=3.2.0
click>=8.1.7
cmake>=3.27.2
CoLT5-attention>=0.10.15
contourpy>=1.1.0
cycler>=0.11.0
datasets>=2.14.4
deepspeed>=0.10.0
dill>=0.3.7
docker-pycreds>=0.4.0
einops>=0.6.1
evaluate>=0.4.0
filelock>=3.12.2
fonttools>=4.42.1
frozenlist>=1.4.0
fsspec>=2023.6.0
gitdb>=4.0.10
GitPython>=3.1.32
hjson>=3.1.0
huggingface-hub>=0.16.4
idna>=3.4
importlib-resources>=6.0.1
Jinja2>=3.1.2
kiwisolver>=1.4.4
lit>=16.0.6
llvmlite>=0.40.1
local-attention>=1.8.6
MarkupSafe>=2.1.3
matplotlib>=3.7.2
mpmath>=1.3.0
multidict>=6.0.4
multiprocess>=0.70.15
networkx>=3.1
ninja>=1.11.1
numba>=0.57.1
numpy>=1.24.4
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>=12.535.77
nvidia-nccl-cu11>=2.14.3
nvidia-nvtx-cu11>=11.7.91
nvitop>=1.2.0
opencv-python>=4.8.0.76
packaging>=23.1
pandas>=2.0.3
pathtools>=0.1.2
performer-pytorch>=1.1.4
Pillow>=10.0.0
product-key-memory>=0.2.9
protobuf>=4.24.1
psutil>=5.9.5
py-cpuinfo>=9.0.0
pyarrow>=12.0.1
pydantic>=1.10.12
pyparsing>=3.0.9
python-dateutil>=2.8.2
pytz>=2023.3
PyYAML>=6.0.1
reformer-pytorch>=1.4.4
regex>=2023.8.8
requests>=2.31.0
responses>=0.18.0
safetensors>=0.3.2
sentry-sdk>=1.29.2
setproctitle>=1.3.2
six>=1.16.0
smmap>=5.0.0
sympy>=1.12
termcolor>=2.3.0
tokenizers>=0.13.3
torch>=2.0.1
torchdata>=0.6.1
torchtext>=0.15.2
torchvision>=0.15.2
tqdm>=4.66.1
transformers>=4.32.0
triton>=2.0.0
typing_extensions>=4.7.1
tzdata>=2023.3
urllib3>=2.0.4
wandb>=0.15.8
xxhash>=3.3.0
yarl>=1.9.2
zipp>=3.16.2