Skip to content

Commit edb1e33

Browse files
fix inits for colqwen 2_5 (#118)
* fix inits * f
1 parent ef1ddad commit edb1e33

File tree

3 files changed

+3
-2
lines changed

3 files changed

+3
-2
lines changed

src/vidore_benchmark/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
CohereAPIRetriever,
88
ColIdefics3Retriever,
99
ColPaliRetriever,
10+
ColQwen2_5_Retriever,
1011
ColQwen2Retriever,
1112
DSEQwen2Retriever,
1213
DummyVisionRetriever,

src/vidore_benchmark/retrievers/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,8 @@
66
from .cohere_api_retriever import CohereAPIRetriever
77
from .colidefics3_retriever import ColIdefics3Retriever
88
from .colpali_retriever import ColPaliRetriever
9-
from .colqwen2_retriever import ColQwen2Retriever
109
from .colqwen2_5_retriever import ColQwen2_5_Retriever
10+
from .colqwen2_retriever import ColQwen2Retriever
1111
from .dse_qwen2_retriever import DSEQwen2Retriever
1212
from .dummy_vision_retriever import DummyVisionRetriever
1313
from .hf_api_retriever import HFEndpointRetriever

src/vidore_benchmark/retrievers/colqwen2_5_retriever.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121

2222

2323
@register_vision_retriever("colqwen2_5")
24-
class ColQwen2_5_Retriever(BaseVisionRetriever):
24+
class ColQwen2_5_Retriever(BaseVisionRetriever): #noqa: N801
2525
"""
2626
ColQwen2.5 retriever that implements the model from "ColPali: Efficient Document Retrieval
2727
with Vision Language Models".

0 commit comments

Comments
 (0)