colpali-vespa-visual-retrieval / requirements.txt
thomasht86's picture
Upload folder using huggingface_hub
e52f5a6 verified
raw
history blame
8.9 kB
# This file was autogenerated by uv via the following command:
# uv pip compile pyproject.toml -o requirements.txt
accelerate==0.34.2
# via peft
aiohappyeyeballs==2.4.3
# via aiohttp
aiohttp==3.10.9
# via
# datasets
# fsspec
# pyvespa
aiosignal==1.3.1
# via aiohttp
annotated-types==0.7.0
# via pydantic
anyio==4.6.0
# via
# httpx
# starlette
# watchfiles
async-timeout==4.0.3
# via aiohttp
attrs==24.2.0
# via aiohttp
beautifulsoup4==4.12.3
# via python-fasthtml
cachetools==5.5.0
# via google-auth
certifi==2024.8.30
# via
# httpcore
# httpx
# requests
cffi==1.17.1
# via cryptography
charset-normalizer==3.3.2
# via requests
click==8.1.7
# via
# typer
# uvicorn
colpali-engine==0.3.1
# via
# visual-retrieval-colpali (pyproject.toml)
# vidore-benchmark
contourpy==1.3.0
# via matplotlib
cryptography==43.0.1
# via pyvespa
cycler==0.12.1
# via matplotlib
datasets==2.21.0
# via
# mteb
# vidore-benchmark
dill==0.3.8
# via
# datasets
# multiprocess
docker==7.1.0
# via pyvespa
einops==0.8.0
# via
# visual-retrieval-colpali (pyproject.toml)
# vidore-benchmark
eval-type-backport==0.2.0
# via mteb
exceptiongroup==1.2.2
# via anyio
fastcore==1.7.11
# via
# fastlite
# python-fasthtml
# pyvespa
# sqlite-minutils
fastlite==0.0.11
# via python-fasthtml
filelock==3.16.1
# via
# datasets
# huggingface-hub
# torch
# transformers
fonttools==4.54.1
# via matplotlib
frozenlist==1.4.1
# via
# aiohttp
# aiosignal
fsspec==2024.6.1
# via
# datasets
# huggingface-hub
# torch
google-ai-generativelanguage==0.6.10
# via google-generativeai
google-api-core==2.21.0
# via
# google-ai-generativelanguage
# google-api-python-client
# google-generativeai
google-api-python-client==2.149.0
# via google-generativeai
google-auth==2.35.0
# via
# google-ai-generativelanguage
# google-api-core
# google-api-python-client
# google-auth-httplib2
# google-generativeai
google-auth-httplib2==0.2.0
# via google-api-python-client
google-generativeai==0.8.3
# via visual-retrieval-colpali (pyproject.toml)
googleapis-common-protos==1.65.0
# via
# google-api-core
# grpcio-status
gputil==1.4.0
# via
# colpali-engine
# vidore-benchmark
grpcio==1.67.0
# via
# google-api-core
# grpcio-status
grpcio-status==1.67.0
# via google-api-core
h11==0.14.0
# via
# httpcore
# uvicorn
h2==4.1.0
# via httpx
hpack==4.0.0
# via h2
httpcore==1.0.6
# via httpx
httplib2==0.22.0
# via
# google-api-python-client
# google-auth-httplib2
httptools==0.6.1
# via uvicorn
httpx==0.27.2
# via
# python-fasthtml
# pyvespa
huggingface-hub==0.25.1
# via
# visual-retrieval-colpali (pyproject.toml)
# accelerate
# datasets
# peft
# sentence-transformers
# tokenizers
# transformers
hyperframe==6.0.1
# via h2
idna==3.10
# via
# anyio
# httpx
# requests
# yarl
itsdangerous==2.2.0
# via python-fasthtml
jinja2==3.1.4
# via
# pyvespa
# torch
joblib==1.4.2
# via scikit-learn
kiwisolver==1.4.7
# via matplotlib
loguru==0.7.2
# via vidore-benchmark
lucide-fasthtml==0.0.9
# via shad4fast
lxml==5.3.0
# via
# lucide-fasthtml
# pyvespa
markdown-it-py==3.0.0
# via rich
markupsafe==2.1.5
# via jinja2
matplotlib==3.9.2
# via
# seaborn
# vidore-benchmark
mdurl==0.1.2
# via markdown-it-py
mpmath==1.3.0
# via sympy
mteb==1.15.3
# via vidore-benchmark
multidict==6.1.0
# via
# aiohttp
# yarl
multiprocess==0.70.16
# via datasets
networkx==3.3
# via torch
numpy==1.26.4
# via
# accelerate
# colpali-engine
# contourpy
# datasets
# matplotlib
# mteb
# pandas
# peft
# pyarrow
# scikit-learn
# scipy
# seaborn
# transformers
# vidore-benchmark
oauthlib==3.2.2
# via python-fasthtml
packaging==24.1
# via
# accelerate
# datasets
# fastcore
# huggingface-hub
# matplotlib
# peft
# transformers
pandas==2.2.3
# via
# datasets
# seaborn
pdf2image==1.17.0
# via vidore-benchmark
peft==0.11.1
# via
# colpali-engine
# vidore-benchmark
pillow==10.4.0
# via
# colpali-engine
# matplotlib
# pdf2image
# sentence-transformers
# vidore-benchmark
polars==1.9.0
# via mteb
proto-plus==1.24.0
# via
# google-ai-generativelanguage
# google-api-core
protobuf==5.28.3
# via
# google-ai-generativelanguage
# google-api-core
# google-generativeai
# googleapis-common-protos
# grpcio-status
# proto-plus
psutil==6.0.0
# via
# accelerate
# peft
pyarrow==17.0.0
# via datasets
pyasn1==0.6.1
# via
# pyasn1-modules
# rsa
pyasn1-modules==0.4.1
# via google-auth
pycparser==2.22
# via cffi
pydantic==2.9.2
# via
# google-generativeai
# mteb
pydantic-core==2.23.4
# via pydantic
pygments==2.18.0
# via rich
pyparsing==3.1.4
# via
# httplib2
# matplotlib
pypdf==5.0.1
# via visual-retrieval-colpali (pyproject.toml)
python-dateutil==2.9.0.post0
# via
# matplotlib
# pandas
# python-fasthtml
# pyvespa
python-dotenv==1.0.1
# via
# visual-retrieval-colpali (pyproject.toml)
# uvicorn
# vidore-benchmark
python-fasthtml==0.6.9
# via
# visual-retrieval-colpali (pyproject.toml)
# lucide-fasthtml
# shad4fast
python-multipart==0.0.12
# via python-fasthtml
pytrec-eval-terrier==0.5.6
# via mteb
pytz==2024.2
# via pandas
pyvespa==0.50.0
# via visual-retrieval-colpali (pyproject.toml)
pyyaml==6.0.2
# via
# accelerate
# datasets
# huggingface-hub
# peft
# transformers
# uvicorn
regex==2024.9.11
# via transformers
requests==2.32.3
# via
# colpali-engine
# datasets
# docker
# google-api-core
# huggingface-hub
# lucide-fasthtml
# mteb
# pyvespa
# requests-toolbelt
# transformers
requests-toolbelt==1.0.0
# via pyvespa
rich==13.9.2
# via
# mteb
# typer
rsa==4.9
# via google-auth
safetensors==0.4.5
# via
# accelerate
# peft
# transformers
scikit-learn==1.5.2
# via
# mteb
# sentence-transformers
scipy==1.14.1
# via
# mteb
# scikit-learn
# sentence-transformers
seaborn==0.13.2
# via vidore-benchmark
sentence-transformers==3.1.1
# via
# mteb
# vidore-benchmark
sentencepiece==0.2.0
# via vidore-benchmark
setuptools==75.1.0
# via visual-retrieval-colpali (pyproject.toml)
shad4fast==1.2.1
# via visual-retrieval-colpali (pyproject.toml)
shellingham==1.5.4
# via typer
six==1.16.0
# via python-dateutil
sniffio==1.3.1
# via
# anyio
# httpx
soupsieve==2.6
# via beautifulsoup4
sqlite-minutils==3.37.0.post3
# via fastlite
starlette==0.39.2
# via python-fasthtml
sympy==1.13.3
# via torch
tenacity==9.0.0
# via pyvespa
threadpoolctl==3.5.0
# via scikit-learn
tokenizers==0.20.0
# via transformers
torch==2.4.1
# via
# visual-retrieval-colpali (pyproject.toml)
# accelerate
# colpali-engine
# mteb
# peft
# sentence-transformers
# vidore-benchmark
tqdm==4.66.5
# via
# datasets
# google-generativeai
# huggingface-hub
# mteb
# peft
# sentence-transformers
# transformers
transformers==4.45.1
# via
# colpali-engine
# peft
# sentence-transformers
# vidore-benchmark
typer==0.12.5
# via vidore-benchmark
typing-extensions==4.12.2
# via
# anyio
# google-generativeai
# huggingface-hub
# mteb
# multidict
# pydantic
# pydantic-core
# pypdf
# pyvespa
# rich
# torch
# typer
# uvicorn
tzdata==2024.2
# via pandas
uritemplate==4.1.1
# via google-api-python-client
urllib3==2.2.3
# via
# docker
# requests
uvicorn==0.31.0
# via python-fasthtml
uvloop==0.20.0
# via uvicorn
vespacli==8.391.23
# via visual-retrieval-colpali (pyproject.toml)
vidore-benchmark==4.0.0
# via visual-retrieval-colpali (pyproject.toml)
watchfiles==0.24.0
# via uvicorn
websockets==13.1
# via uvicorn
xxhash==3.5.0
# via datasets
yarl==1.13.1
# via aiohttp