Wheelodex » Projects » colpali-engine

colpali-engine

View on PyPI — Reverse Dependencies (7)

0.3.8 colpali_engine-0.3.8-py3-none-any.whl

Wheel Details

Project: colpali-engine
Version: 0.3.8
Filename: colpali_engine-0.3.8-py3-none-any.whl
Download: [link]
Size: 48525
MD5: fa7246d5603ef26b0cd7b7fd777941d3
SHA256: 6f1bb5b0f0b05c49645962ab6882c9962bf035656202a01cd393d6c1e46d6a93
Uploaded: 2025-01-29 09:18:21 +0000

dist-info

METADATA · WHEEL · RECORD

METADATA

Metadata-Version: 2.4
Name: colpali_engine
Version: 0.3.8
Summary: The code used to train and run inference with the ColPali architecture.
Author-Email: Manuel Faysse <manuel.faysse[at]illuin.tech>, Hugues Sibille <hugues.sibille[at]illuin.tech>, Tony Wu <tony.wu[at]illuin.tech>
Maintainer-Email: Manuel Faysse <manuel.faysse[at]illuin.tech>, Tony Wu <tony.wu[at]illuin.tech>
Project-Url: homepage, https://github.com/illuin-tech/colpali
Classifier: Intended Audience :: Developers
Classifier: Intended Audience :: Science/Research
Classifier: License :: OSI Approved :: MIT License
Classifier: Operating System :: OS Independent
Classifier: Programming Language :: Python :: 3
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Requires-Python: >=3.9
Requires-Dist: gputil
Requires-Dist: numpy
Requires-Dist: peft (<0.15.0,>=0.14.0)
Requires-Dist: pillow (>=10.0.0)
Requires-Dist: requests
Requires-Dist: torch (>=2.2.0)
Requires-Dist: transformers (<4.48.0,>=4.47.0)
Requires-Dist: accelerate (<1.4.0,>=0.34.0); extra == "all"
Requires-Dist: bitsandbytes; extra == "all"
Requires-Dist: configue (>=5.0.0); extra == "all"
Requires-Dist: datasets (>=2.19.1); extra == "all"
Requires-Dist: einops (<1.0.0,>=0.8.0); extra == "all"
Requires-Dist: matplotlib (<4.0.0,>=3.9.0); extra == "all"
Requires-Dist: mteb (<1.17.0,>=1.16.3); extra == "all"
Requires-Dist: pillow (<11.1.0,>=10.0.0); extra == "all"
Requires-Dist: pytest (>=8.0.0); extra == "all"
Requires-Dist: ruff (>=0.4.0); extra == "all"
Requires-Dist: seaborn (<1.0.0,>=0.13.2); extra == "all"
Requires-Dist: typer (>=0.15.1); extra == "all"
Requires-Dist: datasets (>=2.19.1); extra == "dev"
Requires-Dist: pytest (>=8.0.0); extra == "dev"
Requires-Dist: ruff (>=0.4.0); extra == "dev"
Requires-Dist: einops (<1.0.0,>=0.8.0); extra == "interpretability"
Requires-Dist: matplotlib (<4.0.0,>=3.9.0); extra == "interpretability"
Requires-Dist: seaborn (<1.0.0,>=0.13.2); extra == "interpretability"
Requires-Dist: accelerate (<1.4.0,>=0.34.0); extra == "train"
Requires-Dist: bitsandbytes; extra == "train"
Requires-Dist: configue (>=5.0.0); extra == "train"
Requires-Dist: datasets (>=2.19.1); extra == "train"
Requires-Dist: mteb (<1.17.0,>=1.16.3); extra == "train"
Requires-Dist: pillow (<11.1.0,>=10.0.0); extra == "train"
Requires-Dist: typer (>=0.15.1); extra == "train"
Provides-Extra: all
Provides-Extra: dev
Provides-Extra: interpretability
Provides-Extra: train
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 24566 characters]

WHEEL

Wheel-Version: 1.0
Generator: hatchling 1.27.0
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
colpali_engine/__init__.py sha256=_L8qUX5nHTA7r7T0h-dxaeBCWcte3ZR9GiLEy7NLG9c 261
colpali_engine/collators/__init__.py sha256=XdBSPACKUF6Q74Kh_chu03ONLOcqhu41skNJl7lP_OE 118
colpali_engine/collators/corpus_query_collator.py sha256=e1Zo5IopFQo2cKokS0uoYqZTahKA8HbHHyJJiyMIJG8 3395
colpali_engine/collators/visual_retriever_collator.py sha256=NKE8ryyXDoKqXcEJ53wDM-Khe02uacbGZQ9yb7VfThM 3784
colpali_engine/interpretability/__init__.py sha256=bLcyBXQpGTfYbNYvncNjyqAeD6yBGVFhkiDRInddIgQ 197
colpali_engine/interpretability/similarity_map_utils.py sha256=VKJd-9NxHWi3yWPcSidrxPO2lQzZygax3KEwbhXikY8 3449
colpali_engine/interpretability/similarity_maps.py sha256=ac2MTDDq0wiq85F-brI1SySMnx7ZvtFBCElcZn35Zhw 4904
colpali_engine/loss/__init__.py sha256=CPsO2e5jo4No1FEU-ubICZJiR6o8IM25Qm6da1Aumgc 226
colpali_engine/loss/bi_encoder_losses.py sha256=-TNOwP3Wf9w1ADd070mQUQnmnYjdsg9d1DpMaQkIh8M 2479
colpali_engine/loss/late_interaction_losses.py sha256=OUHRftvL-C0V4bRd_cN-fGhJD3QN96JIcqH7RNXFN0A 4748
colpali_engine/models/__init__.py sha256=ZT5enadB_S8olhh0dhg85DMl1fbyAe_pA8Hs0AmDaHQ 285
colpali_engine/models/idefics3/__init__.py sha256=JEZC97n_F1l_GZNQidv2LANMsM7OYz1pVGKdYFRJryA 60
colpali_engine/models/idefics3/colidefics3/__init__.py sha256=32MwpAool1xCtmEdlgVgrulo0z2s_IHw_6aicM-aK8o 103
colpali_engine/models/idefics3/colidefics3/modeling_colidefics3.py sha256=0yBeIVEIJH-r5A6ZPEaqRLrywMSeOx-aj7cpOFYWVdk 1169
colpali_engine/models/idefics3/colidefics3/processing_colidefics3.py sha256=8AdEJnm8lpNhVT3OzpxmOBSPAu0eT_2vBw0rwHLcCQ0 2864
colpali_engine/models/idefics_2/__init__.py sha256=3LrI5QMnMzbaaGujQWmoht3FMFkVrJ4L2lX99zKNgwI 96
colpali_engine/models/idefics_2/biidefics_2/__init__.py sha256=stsgV9_zPdUvhdwdE74UF9AlcS2qnL0dADZggAbr0Z0 45
colpali_engine/models/idefics_2/biidefics_2/modeling_biidefics_2.py sha256=9QJFf6mdmAx26Zbq0tMCCWE7D3qE0aJDegCPcHtWsKY 1044
colpali_engine/models/idefics_2/biidefics_2/processing_biidefics_2.py sha256=t8hEAAU-OmomDJWD0UZJWqxxMXI8aFwjYaghCY_32Dc 758
colpali_engine/models/idefics_2/colidefics_2/__init__.py sha256=PCyjhyfA1suOx5U6_OZYbnjpMNNFA8tFwXsC8lkwLmg 105
colpali_engine/models/idefics_2/colidefics_2/modeling_colidefics_2.py sha256=meaxZPUU3-VjVwO0M0Nrems2ftCTtGDBSgItJcpJQXw 1169
colpali_engine/models/idefics_2/colidefics_2/processing_colidefics_2.py sha256=5w7tp0zWmjg6N-BGZiZJWeW8yrMEYxtSIVvphUcHG_w 2951
colpali_engine/models/paligemma/__init__.py sha256=rQ09D9Q6QsounVzEwA5gbo1ROC8uxWjMA1S3mm7_Rmw 103
colpali_engine/models/paligemma/bipali/__init__.py sha256=42n7Z5YVezM-4lLvRoAPO8JH-iEsi9IOyxNItMyE56k 95
colpali_engine/models/paligemma/bipali/modeling_bipali.py sha256=pIyggeMiN2_9d7PwX1Csl-Xl0LhcDNs7X007KN02734 5653
colpali_engine/models/paligemma/bipali/processing_bipali.py sha256=MDOyIECirC-hinHwjgqv2xHy5unSLV8YPIJLtJe2sdI 728
colpali_engine/models/paligemma/colpali/__init__.py sha256=ySW4-7j1h0Z3Amkyz7B-lTYn1yzTG0_h2RsCTtD-qGY 87
colpali_engine/models/paligemma/colpali/modeling_colpali.py sha256=5GX1yNccVKMs3BBdBPjh2ILPFORFWKk_Z2BH8kGpWpI 3346
colpali_engine/models/paligemma/colpali/processing_colpali.py sha256=QSYzZ1PTZMV6qWYHXGrwGWIHMvCbg-hi9UPRqI_rqaU 3006
colpali_engine/models/qwen2/__init__.py sha256=VkwuJnU1DYWbAXJWbtplyz1PV8lwOkCmKyKbZ09o8kY 97
colpali_engine/models/qwen2/biqwen2/__init__.py sha256=P1N0PqiPSN5fjYjhjME1To_PVL5Jt0O_MNYFhuOILSg 87
colpali_engine/models/qwen2/biqwen2/modeling_biqwen2.py sha256=r4r4uwTkvXb41gFCFVcX-3hwqNDCr_6LphNw1jDPzBQ 4553
colpali_engine/models/qwen2/biqwen2/processing_biqwen2.py sha256=2X-NK8Gj2IAY9eQaJGHpvvrViiPcjtTfbEaVpuCt9cc 1282
colpali_engine/models/qwen2/colqwen2/__init__.py sha256=AoslFmUxyNK_MnXJ2dFgdnuDSDU7TYKLXKfqWbq2Fbo 91
colpali_engine/models/qwen2/colqwen2/modeling_colqwen2.py sha256=3BSHwhBVrk2hQCXL4qEzytZhKuZ_Ge4PLEECFDA76NU 4929
colpali_engine/models/qwen2/colqwen2/processing_colqwen2.py sha256=hIqev_60AztFT1F5aPVWgfJWIwkHlPxavm5t-S5ZJ_o 7108
colpali_engine/trainer/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
colpali_engine/trainer/colmodel_training.py sha256=wgjKHj3w5ZP6zzoAi6JSiUBixyAGrFabuwGT-kUq9qM 10384
colpali_engine/trainer/contrastive_trainer.py sha256=QUBJHWpShnDBPUbnSDLjT-RV9vOjEWu8EA3srcFkrzQ 2120
colpali_engine/trainer/eval_utils.py sha256=6rVZXqE26KN7GEL3r4nYVCmQkV46yuHcPPscN-kZY9E 6802
colpali_engine/utils/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
colpali_engine/utils/dataset_transformation.py sha256=-uOu-ivkyqriXJIsWBTu06MzpW1QC1XFbhzSDH7qUeI 8474
colpali_engine/utils/gpu_stats.py sha256=1XKZKYIr2miR5YYxjx-Jgwg3zrCPhPEeuWUGdkxtTCU 835
colpali_engine/utils/processing_utils.py sha256=i2X7DUP6honH08FcE6bdg1E3UaA_OEXZ3w-ffwVNZaM 4775
colpali_engine/utils/torch_utils.py sha256=SsaXOghWKq-EuZr-9h2l2T3Lu5GMwx7igBwQQhcNmvk 1234
colpali_engine/utils/transformers_wrappers.py sha256=Bt0ehqtCH5qh5CFja0GhBto7EaUh1FJXHbWcz_LmVvA 742
colpali_engine-0.3.8.dist-info/METADATA sha256=Gp7KmTGb40GL5cXhCHbGzLsnMKNlpW-jhtTyuLrfOFI 27129
colpali_engine-0.3.8.dist-info/WHEEL sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ 87
colpali_engine-0.3.8.dist-info/licenses/LICENSE sha256=AJcdXyMp53jwUIXuvz8Kct1ia3tEj4yFSZl0IMxjOTI 1095
colpali_engine-0.3.8.dist-info/RECORD — —