Add ColQwen2_5 to IMAGE_COL_MODELS (#568)

yasyf · web-flow · commit 0bbfdbc2fa13 · 2025-04-16T10:01:38.000-07:00
* Add ColQwen2_5 to IMAGE_COL_MODELS

* Add ColQwen2_5 classes.
diff --git a/libs/infinity_emb/infinity_emb/transformer/vision/__init__.py b/libs/infinity_emb/infinity_emb/transformer/vision/__init__.py
@@ -4,6 +4,7 @@
 IMAGE_COL_MODELS = [
     "ColPali",
     "ColQwen2",
+    "ColQwen2_5",
     "ColIdefics2",
 ]
 
diff --git a/libs/infinity_emb/infinity_emb/transformer/vision/torch_vision.py b/libs/infinity_emb/infinity_emb/transformer/vision/torch_vision.py
@@ -66,16 +66,20 @@ def __init__(self, *, engine_args: "EngineArgs"):
                 ColPaliProcessor,
                 ColQwen2,
                 ColQwen2Processor,
+                ColQwen2_5, 
+                ColQwen2_5_Processor
             )
 
             model_cls = {
                 "ColPali": ColPali,
                 "ColQwen2": ColQwen2,
+                "ColQwen2_5": ColQwen2_5,
                 "ColIdefics2": ColIdefics2,
             }[config.architectures[0]]
             processor_cls = {
                 "ColPali": ColPaliProcessor,
                 "ColQwen2": ColQwen2Processor,
+                "ColQwen2_5": ColQwen2_5_Processor,
                 "ColIdefics2": ColIdefics2Processor,
             }[config.architectures[0]]
 
