Merge pull request #924 from vespa-engine/jobergum/use-correct-device

Use correct device
vespa-engine · Sep 18, 2024 · 22a0ead · 22a0ead
2 parents 1e481f1 + d926730
commit 22a0ead
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
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
@@ -333,6 +333,7 @@
                 "model = ColPali.from_pretrained(\"vidore/colpaligemma-3b-mix-448-base\", torch_dtype=type).eval()\n",
                 "model.load_adapter(model_name)\n",
                 "model = model.eval()\n",
+                "model.to(device)\n",
                 "processor = AutoProcessor.from_pretrained(model_name)"
             ]
         },

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
@@ -326,6 +326,7 @@
     "model = ColPali.from_pretrained(\"vidore/colpaligemma-3b-mix-448-base\", torch_dtype=type).eval()\n",
     "model.load_adapter(model_name)\n",
     "model = model.eval()\n",
+    "model.to(device)\n",
     "processor = AutoProcessor.from_pretrained(model_name)"
    ]
   },