diff --git a/notebooks/data_gemma_rag.ipynb b/notebooks/data_gemma_rag.ipynb index 40347d6..950121a 100644 --- 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 @@ } } ] -} \ No newline at end of file +}