Add frozen evaluation environment requirements

evaluation-requirements-cu128.txt

@@ -0,0 +1,183 @@
+# Python 3.12 package environment captured from the ColVec1.1 SDPA evaluation.
+# Platform: Linux x86-64, PyTorch 2.9.0, CUDA 12.8.
+# Install PyTorch from the CUDA 12.8 index before installing this file.
+# This is the complete evaluation environment, not the minimal inference setup.
+
+absl-py==2.5.0
+accelerate==1.14.0
+aiohappyeyeballs==2.7.1
+aiohttp==3.14.3
+aiohttp-cors==0.8.1
+aiosignal==1.4.0
+annotated-doc==0.0.4
+annotated-types==0.8.0
+antlr4-python3-runtime==4.9.3
+anyio==4.14.2
+apache-tvm-ffi==0.1.9
+attrs==26.1.0
+bitsandbytes==0.50.0
+boto3==1.43.56
+botocore==1.43.56
+braceexpand==0.1.7
+causal-conv1d @ https://github.com/Dao-AILab/causal-conv1d/releases/download/v1.6.2.post1/causal_conv1d-1.6.2.post1+cu12torch2.9cxx11abiTRUE-cp312-cp312-linux_x86_64.whl
+certifi==2026.7.22
+cffi==2.1.0
+charset-normalizer==3.4.9
+click==8.4.2
+cloudpickle==3.1.2
+colorful==0.5.8
+colpali-engine @ git+https://github.com/illuin-tech/colpali@2e0b927051af727238783af039dcc2c50a4d8c27
+contourpy==1.3.3
+cryptography==49.0.0
+cycler==0.12.1
+datasets==5.0.0
+dill==0.4.1
+distlib==0.4.3
+docstring-parser==0.18.0
+einops==0.8.2
+filelock==3.29.0
+fla-core==0.5.1
+flash-attn @ https://github.com/Dao-AILab/flash-attention/releases/download/v2.8.3/flash_attn-2.8.3+cu12torch2.9cxx11abiTRUE-cp312-cp312-linux_x86_64.whl
+flash-linear-attention==0.5.1
+fonttools==4.63.0
+frozenlist==1.8.0
+fsspec==2026.4.0
+gapstatistics==0.1.5
+google-api-core==2.33.0
+google-auth==2.56.2
+googleapis-common-protos==1.75.0
+grpcio==1.83.0
+h11==0.16.0
+h2==4.4.0
+hf-xet==1.5.2
+hpack==4.2.0
+httpcore==1.0.9
+httpx==0.28.1
+huggingface-hub==1.24.0
+hydra-core==1.3.4
+hyperframe==6.1.0
+idna==3.18
+importlib-resources==7.1.0
+iniconfig==2.3.0
+jinja2==3.1.6
+jmespath==1.1.0
+joblib==1.5.3
+jsonargparse==4.50.0
+jsonnet==0.22.0
+jsonschema==4.26.0
+jsonschema-specifications==2025.9.1
+kiwisolver==1.5.0
+lightning==2.6.5
+lightning-utilities==0.15.3
+markdown==3.10.2
+markdown-it-py==4.2.0
+markupsafe==3.0.3
+matplotlib==3.11.1
+mdurl==0.1.2
+ml-dtypes==0.5.4
+mpmath==1.3.0
+msgpack==1.2.1
+mteb[xet] @ git+https://github.com/embeddings-benchmark/mteb.git@d56a414b45ebad0d03495de000b4880d8b028d4a
+multidict==6.7.1
+multiprocess==0.70.19
+narwhals==2.24.0
+networkx==3.6.1
+numpy==2.4.4
+nvidia-cublas-cu12==12.8.4.1
+nvidia-cuda-cupti-cu12==12.8.90
+nvidia-cuda-nvrtc-cu12==12.8.93
+nvidia-cuda-runtime-cu12==12.8.90
+nvidia-cudnn-cu12==9.10.2.21
+nvidia-cufft-cu12==11.3.3.83
+nvidia-cufile-cu12==1.13.1.3
+nvidia-curand-cu12==10.3.9.90
+nvidia-cusolver-cu12==11.7.3.90
+nvidia-cusparse-cu12==12.5.8.93
+nvidia-cusparselt-cu12==0.7.1
+nvidia-nccl-cu12==2.27.5
+nvidia-nvjitlink-cu12==12.8.93
+nvidia-nvshmem-cu12==3.3.20
+nvidia-nvtx-cu12==12.8.90
+omegaconf==2.3.1
+opencensus==0.11.4
+opencensus-context==0.1.3
+opentelemetry-api==1.44.0
+opentelemetry-exporter-prometheus==0.65b0
+opentelemetry-proto==1.44.0
+opentelemetry-sdk==1.44.0
+opentelemetry-semantic-conventions==0.65b0
+packaging==26.2
+pandas==3.0.5
+peft==0.19.1
+pillow==12.2.0
+platformdirs==4.11.0
+pluggy==1.6.0
+polars==1.43.0
+polars-runtime-32==1.43.0
+portalocker==3.2.0
+prometheus-client==0.26.0
+propcache==0.5.2
+proto-plus==1.28.2
+protobuf==7.35.1
+psutil==7.2.2
+py-spy==0.4.2
+pyarrow==25.0.0
+pyasn1==0.6.4
+pyasn1-modules==0.4.2
+pycparser==3.0
+pydantic==2.13.4
+pydantic-core==2.46.4
+pygments==2.20.0
+pymupdf==1.28.0
+pyparsing==3.3.2
+pytest==9.1.1
+python-dateutil==2.9.0.post0
+python-discovery==1.5.0
+python-dotenv==1.2.2
+pytorch-lightning==2.6.5
+pytrec-eval-terrier==0.5.10
+pyyaml==6.0.3
+qdrant-client==1.18.0
+ray==2.56.1
+referencing==0.37.0
+regex==2026.7.19
+requests==2.34.2
+rich==14.3.4
+rpds-py==2026.6.3
+s3transfer==0.19.2
+safetensors==0.8.0
+scikit-learn==1.9.0
+scipy==1.18.0
+sentence-transformers==5.6.0
+sentry-sdk==2.66.1
+setuptools==78.1.0
+shellingham==1.5.4
+six==1.17.0
+smart-open==8.0.1
+sympy==1.14.0
+tensorboard==2.21.0
+tensorboard-data-server==0.7.2
+tensorboardx==2.6.5
+threadpoolctl==3.6.0
+tilelang==0.1.9
+tokenizers==0.22.2
+torch==2.9.0+cu128
+torch-c-dlpack-ext==0.1.5
+torchmetrics==1.9.0
+torchvision==0.24.0+cu128
+tqdm==4.69.1
+transformers==5.14.1
+triton==3.5.0
+typer==0.27.0
+typeshed-client==2.12.0
+typing-extensions==4.15.0
+typing-inspection==0.4.2
+urllib3==2.7.0
+virtualenv==21.7.0
+wandb==0.28.1
+webdataset==1.0.2
+werkzeug==3.1.8
+wrapt==2.2.2
+xxhash==3.8.1
+yarl==1.24.5
+z3-solver==4.15.4.0