add deps (#570)

* add deps

* revert torch vision

Author: michaelfeil

libs/infinity_emb/infinity_emb/transformer/vision/torch_vision.py

@@ -66,20 +66,21 @@ def __init__(self, *, engine_args: "EngineArgs"):
                 ColPaliProcessor,
                 ColQwen2,
                 ColQwen2Processor,
-                ColQwen2_5, 
-                ColQwen2_5_Processor
+                # TODO: colqwen 2.5 is not supported with colpali-engine 0.3.8
+                # ColQwen2_5,
+                # ColQwen2_5_Processor,
             )
 
             model_cls = {
                 "ColPali": ColPali,
                 "ColQwen2": ColQwen2,
-                "ColQwen2_5": ColQwen2_5,
+                # "ColQwen2_5": ColQwen2_5,
                 "ColIdefics2": ColIdefics2,
             }[config.architectures[0]]
             processor_cls = {
                 "ColPali": ColPaliProcessor,
                 "ColQwen2": ColQwen2Processor,
-                "ColQwen2_5": ColQwen2_5_Processor,
+                # "ColQwen2_5": ColQwen2_5_Processor,
                 "ColIdefics2": ColIdefics2Processor,
             }[config.architectures[0]]
 

libs/infinity_emb/poetry.lock

@@ -30,15 +30,15 @@ posthog = {version = "*", optional=true}
 # pin torch to a specific source, but default to pypi. use sed to overwrite.
 torch = {version = ">=2.2.1", source = "pypi", optional=true}
 sentence-transformers = {version = "^3.0.1", optional=true} 
-transformers = {version = ">4.34.0,<=5.0", optional=true, extras=["sentencepiece"]}
-ctranslate2 = {version = "^4.0.0", optional=true}
-optimum = {version = ">=1.23.3", optional=true, extras=["onnxruntime"]}
+transformers = {version = ">=4.47.0,<=5.0", optional=true, extras=["sentencepiece"]}
+ctranslate2 = {version = ">=4.0.0", optional=true}
+optimum = {version = ">=1.24.0", optional=true, extras=["onnxruntime"]}
 hf_transfer = {version=">=0.1.5"}
 einops = {version = "*", optional=true}
 # vision
 pillow = {version = "*", optional=true}
 timm = {version = "*", optional=true}
-colpali-engine = {version="^0.3.1", optional=true}
+colpali-engine = {version="^0.3.8", optional=true}
 # openvino
 # optimum-intel = {version=">=1.20.0", optional=true, extras=["openvino"]}
 # onnxruntime-openvino = {version=">=1.19.0", optional=true}