From 9d1cb6d876d79f1116fbdf0c3854d120005cc15b Mon Sep 17 00:00:00 2001 From: thomasht86 Date: Thu, 21 Nov 2024 16:32:01 +0100 Subject: [PATCH] dtype --- .../examples/colpali-benchmark-vqa-vlm_Vespa-cloud.ipynb | 2 +- ...ali-document-retrieval-vision-language-models-cloud.ipynb | 2 +- .../simplified-retrieval-with-colpali-vlm_Vespa-cloud.ipynb | 5 +++-- 3 files changed, 5 insertions(+), 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 index da59d118..483229b8 100644 --- 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 index 9cc40752..bc5b6c3b 100644 --- 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 index ad185d43..2867a9ef 100644 --- 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,7 +433,7 @@ }, { "cell_type": "code", - "execution_count": 6, + "execution_count": null, "metadata": { "id": "YaDInfmT3Tbu" }, @@ -441,6 +441,7 @@ "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" ]