reqs

requirements.in

@@ -0,0 +1,5 @@
+jupyterlab==4.2.5
+tornado==6.2
+ipywidgets
+colpali-engine[train]>=0.3.0,<0.4.0
+huggingface_hub

requirements.txt

@@ -1,4 +1,486 @@
+# This file was autogenerated by uv via the following command:
+#    uv pip compile requirements.in -o requirements.txt
+accelerate==0.34.2
+    # via
+    #   colpali-engine
+    #   peft
+aiohappyeyeballs==2.4.0
+    # via aiohttp
+aiohttp==3.10.6
+    # via
+    #   datasets
+    #   fsspec
+aiosignal==1.3.1
+    # via aiohttp
+annotated-types==0.7.0
+    # via pydantic
+anyio==4.6.0
+    # via
+    #   httpx
+    #   jupyter-server
+appnope==0.1.4
+    # via ipykernel
+argon2-cffi==23.1.0
+    # via jupyter-server
+argon2-cffi-bindings==21.2.0
+    # via argon2-cffi
+arrow==1.3.0
+    # via isoduration
+asttokens==2.4.1
+    # via stack-data
+async-lru==2.0.4
+    # via jupyterlab
+attrs==24.2.0
+    # via
+    #   aiohttp
+    #   jsonschema
+    #   referencing
+babel==2.16.0
+    # via jupyterlab-server
+beautifulsoup4==4.12.3
+    # via nbconvert
+bitsandbytes==0.42.0
+    # via colpali-engine
+bleach==6.1.0
+    # via nbconvert
+certifi==2024.8.30
+    # via
+    #   httpcore
+    #   httpx
+    #   requests
+cffi==1.17.1
+    # via argon2-cffi-bindings
+charset-normalizer==3.3.2
+    # via requests
+click==8.1.7
+    # via typer
+colpali-engine==0.3.0
+    # via -r requirements.in
+comm==0.2.2
+    # via
+    #   ipykernel
+    #   ipywidgets
+configue==5.0.0
+    # via colpali-engine
+datasets==3.0.0
+    # via
+    #   colpali-engine
+    #   mteb
+debugpy==1.8.6
+    # via ipykernel
+decorator==5.1.1
+    # via ipython
+defusedxml==0.7.1
+    # via nbconvert
+dill==0.3.8
+    # via
+    #   datasets
+    #   multiprocess
+eval-type-backport==0.2.0
+    # via mteb
+executing==2.1.0
+    # via stack-data
+fastjsonschema==2.20.0
+    # via nbformat
+filelock==3.16.1
+    # via
+    #   datasets
+    #   huggingface-hub
+    #   torch
+    #   transformers
+fqdn==1.5.1
+    # via jsonschema
+frozenlist==1.4.1
+    # via
+    #   aiohttp
+    #   aiosignal
+fsspec==2024.6.1
+    # via
+    #   datasets
+    #   huggingface-hub
+    #   torch
+gputil==1.4.0
+    # via colpali-engine
+h11==0.14.0
+    # via httpcore
+httpcore==1.0.5
+    # via httpx
+httpx==0.27.2
+    # via jupyterlab
+huggingface-hub==0.25.1
+    # via
+    #   -r requirements.in
+    #   accelerate
+    #   datasets
+    #   peft
+    #   sentence-transformers
+    #   tokenizers
+    #   transformers
+idna==3.10
+    # via
+    #   anyio
+    #   httpx
+    #   jsonschema
+    #   requests
+    #   yarl
+ipykernel==6.29.5
+    # via jupyterlab
+ipython==8.27.0
+    # via
+    #   ipykernel
+    #   ipywidgets
+ipywidgets==8.1.5
+    # via -r requirements.in
+isoduration==20.11.0
+    # via jsonschema
+jedi==0.19.1
+    # via ipython
+jinja2==3.1.4
+    # via
+    #   jupyter-server
+    #   jupyterlab
+    #   jupyterlab-server
+    #   nbconvert
+    #   torch
+joblib==1.4.2
+    # via scikit-learn
+json5==0.9.25
+    # via jupyterlab-server
+jsonpointer==3.0.0
+    # via jsonschema
+jsonschema==4.23.0
+    # via
+    #   jupyter-events
+    #   jupyterlab-server
+    #   nbformat
+jsonschema-specifications==2023.12.1
+    # via jsonschema
+jupyter-client==8.6.3
+    # via
+    #   ipykernel
+    #   jupyter-server
+    #   nbclient
+jupyter-core==5.7.2
+    # via
+    #   ipykernel
+    #   jupyter-client
+    #   jupyter-server
+    #   jupyterlab
+    #   nbclient
+    #   nbconvert
+    #   nbformat
+jupyter-events==0.10.0
+    # via jupyter-server
+jupyter-lsp==2.2.5
+    # via jupyterlab
+jupyter-server==2.14.2
+    # via
+    #   jupyter-lsp
+    #   jupyterlab
+    #   jupyterlab-server
+    #   notebook-shim
+jupyter-server-terminals==0.5.3
+    # via jupyter-server
 jupyterlab==4.2.5
+    # via -r requirements.in
+jupyterlab-pygments==0.3.0
+    # via nbconvert
+jupyterlab-server==2.27.3
+    # via jupyterlab
+jupyterlab-widgets==3.0.13
+    # via ipywidgets
+markdown-it-py==3.0.0
+    # via rich
+markupsafe==2.1.5
+    # via
+    #   jinja2
+    #   nbconvert
+matplotlib-inline==0.1.7
+    # via
+    #   ipykernel
+    #   ipython
+mdurl==0.1.2
+    # via markdown-it-py
+mistune==3.0.2
+    # via nbconvert
+mpmath==1.3.0
+    # via sympy
+mteb==1.14.21
+    # via colpali-engine
+multidict==6.1.0
+    # via
+    #   aiohttp
+    #   yarl
+multiprocess==0.70.16
+    # via datasets
+nbclient==0.10.0
+    # via nbconvert
+nbconvert==7.16.4
+    # via jupyter-server
+nbformat==5.10.4
+    # via
+    #   jupyter-server
+    #   nbclient
+    #   nbconvert
+nest-asyncio==1.6.0
+    # via ipykernel
+networkx==3.3
+    # via torch
+notebook-shim==0.2.4
+    # via jupyterlab
+numpy==1.26.4
+    # via
+    #   accelerate
+    #   colpali-engine
+    #   datasets
+    #   mteb
+    #   pandas
+    #   peft
+    #   pyarrow
+    #   scikit-learn
+    #   scipy
+    #   transformers
+overrides==7.7.0
+    # via jupyter-server
+packaging==24.1
+    # via
+    #   accelerate
+    #   datasets
+    #   huggingface-hub
+    #   ipykernel
+    #   jupyter-server
+    #   jupyterlab
+    #   jupyterlab-server
+    #   nbconvert
+    #   peft
+    #   transformers
+pandas==2.2.3
+    # via datasets
+pandocfilters==1.5.1
+    # via nbconvert
+parso==0.8.4
+    # via jedi
+peft==0.11.1
+    # via colpali-engine
+pexpect==4.9.0
+    # via ipython
+pillow==10.4.0
+    # via
+    #   colpali-engine
+    #   sentence-transformers
+platformdirs==4.3.6
+    # via jupyter-core
+polars==1.8.2
+    # via mteb
+prometheus-client==0.21.0
+    # via jupyter-server
+prompt-toolkit==3.0.48
+    # via ipython
+psutil==6.0.0
+    # via
+    #   accelerate
+    #   ipykernel
+    #   peft
+ptyprocess==0.7.0
+    # via
+    #   pexpect
+    #   terminado
+pure-eval==0.2.3
+    # via stack-data
+pyarrow==17.0.0
+    # via datasets
+pycparser==2.22
+    # via cffi
+pydantic==2.9.2
+    # via mteb
+pydantic-core==2.23.4
+    # via pydantic
+pygments==2.18.0
+    # via
+    #   ipython
+    #   nbconvert
+    #   rich
+python-dateutil==2.9.0.post0
+    # via
+    #   arrow
+    #   jupyter-client
+    #   pandas
+python-json-logger==2.0.7
+    # via jupyter-events
+pytrec-eval-terrier==0.5.6
+    # via mteb
+pytz==2024.2
+    # via pandas
+pyyaml==6.0.2
+    # via
+    #   accelerate
+    #   configue
+    #   datasets
+    #   huggingface-hub
+    #   jupyter-events
+    #   peft
+    #   transformers
+pyzmq==26.2.0
+    # via
+    #   ipykernel
+    #   jupyter-client
+    #   jupyter-server
+referencing==0.35.1
+    # via
+    #   jsonschema
+    #   jsonschema-specifications
+    #   jupyter-events
+regex==2024.9.11
+    # via transformers
+requests==2.32.3
+    # via
+    #   colpali-engine
+    #   datasets
+    #   huggingface-hub
+    #   jupyterlab-server
+    #   mteb
+    #   transformers
+rfc3339-validator==0.1.4
+    # via
+    #   jsonschema
+    #   jupyter-events
+rfc3986-validator==0.1.1
+    # via
+    #   jsonschema
+    #   jupyter-events
+rich==13.8.1
+    # via
+    #   mteb
+    #   typer
+rpds-py==0.20.0
+    # via
+    #   jsonschema
+    #   referencing
+safetensors==0.4.5
+    # via
+    #   accelerate
+    #   peft
+    #   transformers
+scikit-learn==1.5.2
+    # via
+    #   mteb
+    #   sentence-transformers
+scipy==1.14.1
+    # via
+    #   bitsandbytes
+    #   mteb
+    #   scikit-learn
+    #   sentence-transformers
+send2trash==1.8.3
+    # via jupyter-server
+sentence-transformers==3.1.1
+    # via mteb
+setuptools==75.1.0
+    # via
+    #   jupyterlab
+    #   torch
+shellingham==1.5.4
+    # via typer
+six==1.16.0
+    # via
+    #   asttokens
+    #   bleach
+    #   python-dateutil
+    #   rfc3339-validator
+sniffio==1.3.1
+    # via
+    #   anyio
+    #   httpx
+soupsieve==2.6
+    # via beautifulsoup4
+stack-data==0.6.3
+    # via ipython
+sympy==1.13.3
+    # via torch
+terminado==0.18.1
+    # via
+    #   jupyter-server
+    #   jupyter-server-terminals
+threadpoolctl==3.5.0
+    # via scikit-learn
+tinycss2==1.3.0
+    # via nbconvert
+tokenizers==0.19.1
+    # via transformers
+torch==2.4.1
+    # via
+    #   accelerate
+    #   colpali-engine
+    #   mteb
+    #   peft
+    #   sentence-transformers
 tornado==6.2
-ipywidgets
-colpali-engine[train]>=0.3.0,<0.4.0
+    # via
+    #   -r requirements.in
+    #   ipykernel
+    #   jupyter-client
+    #   jupyter-server
+    #   jupyterlab
+    #   terminado
+tqdm==4.66.5
+    # via
+    #   datasets
+    #   huggingface-hub
+    #   mteb
+    #   peft
+    #   sentence-transformers
+    #   transformers
+traitlets==5.14.3
+    # via
+    #   comm
+    #   ipykernel
+    #   ipython
+    #   ipywidgets
+    #   jupyter-client
+    #   jupyter-core
+    #   jupyter-events
+    #   jupyter-server
+    #   jupyterlab
+    #   matplotlib-inline
+    #   nbclient
+    #   nbconvert
+    #   nbformat
+transformers==4.44.2
+    # via
+    #   colpali-engine
+    #   peft
+    #   sentence-transformers
+typer==0.12.5
+    # via colpali-engine
+types-python-dateutil==2.9.0.20240906
+    # via arrow
+typing-extensions==4.12.2
+    # via
+    #   huggingface-hub
+    #   mteb
+    #   pydantic
+    #   pydantic-core
+    #   torch
+    #   typer
+tzdata==2024.2
+    # via pandas
+uri-template==1.3.0
+    # via jsonschema
+urllib3==2.2.3
+    # via requests
+wcwidth==0.2.13
+    # via prompt-toolkit
+webcolors==24.8.0
+    # via jsonschema
+webencodings==0.5.1
+    # via
+    #   bleach
+    #   tinycss2
+websocket-client==1.8.0
+    # via jupyter-server
+widgetsnbextension==4.0.13
+    # via ipywidgets
+xxhash==3.5.0
+    # via datasets
+yarl==1.12.1
+    # via aiohttp