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