update colpali-engine

vespa-engine · Oct 29, 2024 · be2a0e3 · be2a0e3
1 parent 050796c
commit be2a0e3
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/docs/sphinx/source/examples/simplified-retrieval-with-colpali-vlm_Vespa-cloud.ipynb b/docs/sphinx/source/examples/simplified-retrieval-with-colpali-vlm_Vespa-cloud.ipynb
@@ -79,7 +79,7 @@
    },
    "outputs": [],
    "source": [
-    "!pip3 install colpali-engine==0.2.2 pdf2image pypdf==5.0.1 pyvespa vespacli requests numpy"
+    "!pip3 install colpali-engine==0.3.2 pdf2image pypdf==5.0.1 pyvespa vespacli requests numpy"
    ]
   },
   {
@@ -329,9 +329,7 @@
    "outputs": [],
    "source": [
     "model_name = \"vidore/colpali-v1.2\"\n",
-    "model = ColPali.from_pretrained(\n",
-    "    \"vidore/colpaligemma-3b-pt-448-base\", torch_dtype=type\n",
-    ").eval()\n",
+    "model = ColPali.from_pretrained(model_name, torch_dtype=type).eval()\n",
     "model.load_adapter(model_name)\n",
     "model = model.eval()\n",
     "model.to(device)\n",