requirements.txt

absl-py==2.1.0
cachetools==5.5.0
click==8.1.7
git+https://github.com/openai/CLIP.git
cuda-python==12.6.0
cudf-cu12==24.8.2
--extra-index-url https://pypi.nvidia.com cuml-cu12
cupy-cuda12x==13.3.0
dask==2024.7.1
dask-cuda==24.8.2
dask-cudf-cu12==24.8.2
dask-expr==1.1.9
distributed==2024.7.1
distributed-ucxx-cu12==0.39.1
fastrlock==0.8.2
fsspec==2024.6.1
ftfy==6.2.3
grpcio==1.66.1
huggingface-hub==0.24.6
importlib-metadata==8.4.0
joblib==1.4.2
libucx-cu12==1.15.0.post1
llvmlite==0.43.0
locket==1.0.0
markdown==3.7
markdown-it-py==3.0.0
mdurl==0.1.2
msgpack==1.0.8
numba==0.60.0
nvtx==0.2.10
pandas==2.2.2
partd==1.4.2
protobuf==5.28.0
psutil==6.0.0
pyarrow==16.1.0
pygments==2.18.0
pylibraft-cu12==24.8.1
pynvjitlink-cu12==0.3.0
pynvml==11.4.1
python-dateutil==2.9.0.post0
pytz==2024.1
raft-dask-cu12==24.8.1
rapids-dask-dependency==24.8.0
regex==2024.7.24
rich==13.8.0
rmm-cu12==24.8.2
safetensors==0.4.4
scikit-learn==1.5.1
scipy==1.13.1
six==1.16.0
sortedcontainers==2.4.0
tblib==3.0.0
tensorboard==2.17.1
tensorboard-data-server==0.7.2
threadpoolctl==3.5.0
toolz==0.12.1
tornado==6.4.1
treelite==4.3.0
tzdata==2024.1
ucx-py-cu12==0.39.2
ucxx-cu12==0.39.1
wcwidth==0.2.13
werkzeug==3.0.4
zict==3.0.0
zipp==3.20.1