From 361cac60112875d3353f3c1087633b2420435bcd Mon Sep 17 00:00:00 2001 From: saied89 Date: Tue, 15 Apr 2025 23:43:29 +0200 Subject: [PATCH 1/2] move the relevant `improved_model` to device instead of `model` from previous section --- 02_embedding_model.ipynb | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/02_embedding_model.ipynb b/02_embedding_model.ipynb index 01a3eff..a30b44b 100644 --- a/02_embedding_model.ipynb +++ b/02_embedding_model.ipynb @@ -1924,13 +1924,13 @@ }, { "cell_type": "code", - "execution_count": 37, + "execution_count": null, "metadata": {}, "outputs": [], "source": [ "import torch\n", "device = torch.device(\"mps\" if torch.cuda.is_available() else \"cpu\")\n", - "model.to(device)\n", + "improved_model.to(device)\n", "minicorpus_docs = minicorpus_docs.to(device)" ] }, From 16a62a2076b69be4c46dd98b58d01516ef6233d3 Mon Sep 17 00:00:00 2001 From: saied89 Date: Tue, 15 Apr 2025 23:50:19 +0200 Subject: [PATCH 2/2] revert unnessary change --- 02_embedding_model.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/02_embedding_model.ipynb b/02_embedding_model.ipynb index a30b44b..c51fc20 100644 --- a/02_embedding_model.ipynb +++ b/02_embedding_model.ipynb @@ -1924,7 +1924,7 @@ }, { "cell_type": "code", - "execution_count": null, + "execution_count": 37, "metadata": {}, "outputs": [], "source": [