소스 검색

fixing model ids

Beto 6 달 전
부모
커밋
648fb4b803
1개의 변경된 파일2개의 추가작업 그리고 3개의 파일을 삭제
  1. 2 3
      recipes/responsible_ai/llama_guard/llama_guard_text_&_vision_inference.ipynb

+ 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",