Wheelodex » Projects » colpali-engine

colpali-engine

View on PyPI — Reverse Dependencies (7)

0.3.4 colpali_engine-0.3.4-py3-none-any.whl

Wheel Details

Project: colpali-engine
Version: 0.3.4
Filename: colpali_engine-0.3.4-py3-none-any.whl
Download: [link]
Size: 44667
MD5: 655d0cd07b465c4a97b3daeb9602fafd
SHA256: fc5d41aaee79ed63db71a38dedc5fd7dfa110f00d2eab67acfa6126b85a5342f
Uploaded: 2024-11-07 15:44:56 +0000

dist-info

METADATA · WHEEL · RECORD

METADATA

Metadata-Version: 2.3
Name: colpali_engine
Version: 0.3.4
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 (<2.0.0)
Requires-Dist: peft (<0.12.0,>=0.11.0)
Requires-Dist: pillow (<11.0.0,>=9.2.0)
Requires-Dist: requests
Requires-Dist: torch (>=2.2.0)
Requires-Dist: transformers (<4.47.0,>=4.46.1)
Requires-Dist: accelerate (<1.1.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: mteb (<1.17.0,>=1.16.3); extra == "all"
Requires-Dist: pytest (>=8.0.0); extra == "all"
Requires-Dist: ruff (>=0.4.0); extra == "all"
Requires-Dist: typer (<1.0.0,>=0.12.3); extra == "all"
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.1.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: typer (<1.0.0,>=0.12.3); 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: 18912 characters]

WHEEL

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

RECORD

Path Digest Size
colpali_engine/__init__.py sha256=yzGPWlmZPCHkOo9nDzks8ig15s5F4tSiiHlVppXF9oU 218
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=VcCmAxjbBCXmiKwKNsChN-f3tdggvlxhkheMPX8_F30 229
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=M-BPOKKJsP1mOEAX9vxtBMm2sivN-YV-bNOIJFExehQ 4680
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=Q6uO21IRoYBU1cBMfPwxgn42n7W7JPc78JUXmfVKQ80 3651
colpali_engine/utils/torch_utils.py sha256=IjrIpNWOIw93h6HR4YulX6m_YGpOFWk6Cxy5BYgAAZs 1150
colpali_engine/utils/transformers_wrappers.py sha256=Bt0ehqtCH5qh5CFja0GhBto7EaUh1FJXHbWcz_LmVvA 742
colpali_engine-0.3.4.dist-info/METADATA sha256=NG16l8oBmFDuvMyIM51vphVnzFmcA1Z9LnJPU_QFcKE 21162
colpali_engine-0.3.4.dist-info/WHEEL sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY 87
colpali_engine-0.3.4.dist-info/licenses/LICENSE sha256=AJcdXyMp53jwUIXuvz8Kct1ia3tEj4yFSZl0IMxjOTI 1095
colpali_engine-0.3.4.dist-info/RECORD — —