Update data_gemma_rag.ipynb (#17)

datacommonsorg · Sep 10, 2024 · 82562e8 · 82562e8
1 parent 014cc22
commit 82562e8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/notebooks/data_gemma_rag.ipynb b/notebooks/data_gemma_rag.ipynb
@@ -117,7 +117,7 @@
         "   bnb_4bit_compute_dtype=torch.bfloat16\n",
         ")\n",
         "\n",
-        "model_name = 'google/data-gemma-rag-27b-it'\n",
+        "model_name = 'google/datagemma-rag-27b-it'\n",
         "tokenizer = AutoTokenizer.from_pretrained(model_name, token=HF_TOKEN)\n",
         "datagemma_model = AutoModelForCausalLM.from_pretrained(model_name,\n",
         "                                             device_map=\"auto\",\n",
@@ -269,4 +269,4 @@
       }
     }
   ]
-}
+}