dtype

vespa-engine · Nov 21, 2024 · 9d1cb6d · 9d1cb6d
1 parent 8be6036
commit 9d1cb6d
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/docs/sphinx/source/examples/colpali-benchmark-vqa-vlm_Vespa-cloud.ipynb b/docs/sphinx/source/examples/colpali-benchmark-vqa-vlm_Vespa-cloud.ipynb
@@ -287,7 +287,7 @@
    "source": [
     "model_name = \"vidore/colpali-v1.2\"\n",
     "model = ColPali.from_pretrained(\n",
-    "    \"vidore/colpaligemma-3b-pt-448-base\", torch_dtype=type\n",
+    "    \"vidore/colpaligemma-3b-pt-448-base\", torch_dtype=dtype\n",
     ").eval()\n",
     "model.load_adapter(model_name)\n",
     "model = model.eval()\n",

diff --git a/docs/sphinx/source/examples/colpali-document-retrieval-vision-language-models-cloud.ipynb b/docs/sphinx/source/examples/colpali-document-retrieval-vision-language-models-cloud.ipynb
@@ -340,7 +340,7 @@
    "source": [
     "model_name = \"vidore/colpali-v1.2\"\n",
     "model = ColPali.from_pretrained(\n",
-    "    \"vidore/colpaligemma-3b-pt-448-base\", torch_dtype=type\n",
+    "    \"vidore/colpaligemma-3b-pt-448-base\", torch_dtype=dtype\n",
     ").eval()\n",
     "model.load_adapter(model_name)\n",
     "model = model.eval()\n",

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
@@ -332,7 +332,7 @@
    "source": [
     "model_name = \"vidore/colpali-v1.2\"\n",
     "model = ColPali.from_pretrained(\n",
-    "    \"vidore/colpaligemma-3b-pt-448-base\", torch_dtype=type\n",
+    "    \"vidore/colpaligemma-3b-pt-448-base\", torch_dtype=dtype\n",
     ").eval()\n",
     "model.load_adapter(model_name)\n",
     "model = model.eval()\n",
@@ -433,14 +433,15 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 6,
+   "execution_count": null,
    "metadata": {
     "id": "YaDInfmT3Tbu"
    },
    "outputs": [],
    "source": [
     "for pdf in sample_pdfs:\n",
     "    page_images, page_texts = get_pdf_images(pdf[\"url\"])\n",
+    "\n",
     "    pdf[\"images\"] = page_images\n",
     "    pdf[\"texts\"] = page_texts"
    ]