Beto 7 mesi fa
parent
commit
648fb4b803

+ 2 - 3
recipes/responsible_ai/llama_guard/llama_guard_text_&_vision_inference.ipynb

@@ -51,9 +51,8 @@
     "from typing import List, Any\n",
     "import torch\n",
     "\n",
-    "# TODO update to public url after testing is done\n",
-    "lg_small_text_model_id = \"nltpt/Llama-Guard-3-1B\"\n",
-    "lg_mm_model_id = \"nltpt/Llama-Guard-3-11B-Vision\"\n",
+    "lg_small_text_model_id = \"meta-llama/Llama-Guard-3-1B\"\n",
+    "lg_mm_model_id = \"meta-llama/Llama-Guard-3-11B-Vision\"\n",
     "\n",
     "# Loading the 1B text only model\n",
     "lg_small_text_tokenizer = AutoTokenizer.from_pretrained(lg_small_text_model_id)\n",