requirements.txt

absl-py==1.4.0
accelerate==0.20.3
aim==3.17.5
aim-ui==3.17.5
aimrecords==0.0.7
aimrocks==0.4.0
aiofiles==23.1.0
alembic==1.11.1
annotated-types==0.5.0
anyio==3.7.1
backoff==2.2.1
base58==2.0.1
bitsandbytes==0.39.1
cachetools==5.3.1
certifi==2023.5.7
cffi==1.15.1
charset-normalizer==3.2.0
click==8.1.5
cmake==3.26.4
cryptography==41.0.2
einops==0.6.1
exceptiongroup==1.1.2
fastapi==0.100.0
filelock==3.12.2
fsspec==2023.6.0
greenlet==2.0.2
grpcio==1.56.0
h11==0.14.0
huggingface-hub==0.16.4
idna==3.4
Jinja2==3.1.2
joblib==1.3.1
lit==16.0.6
Mako==1.2.4
MarkupSafe==2.1.3
monotonic==1.6
mpmath==1.3.0
networkx==3.1
nltk==3.8.1
numpy==1.25.1
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
Pillow==10.0.0
protobuf==4.23.4
psutil==5.9.5
py3nvml==0.2.7
pycparser==2.21
pydantic==2.0.3
pydantic_core==2.3.0
python-dateutil==2.8.2
pytz==2023.3
PyYAML==6.0
regex==2023.6.3
requests==2.31.0
RestrictedPython==6.1
rouge-score==0.1.2
safetensors==0.3.1
scipy==1.11.1
segment-analytics-python==2.2.3
six==1.16.0
sniffio==1.3.0
SQLAlchemy==1.4.49
starlette==0.27.0
sympy==1.12
tokenizers==0.13.3
torch==2.0.1
tqdm==4.65.0
triton==2.0.0
typing_extensions==4.7.1
urllib3==2.0.3
uvicorn==0.23.0
xmltodict==0.13.0