fix(jailbreak): capitalization of Snowflake in use of snowflake-arctic-embed-m-long name. (#1464)

Signed-off-by: Erick Galinkin <egalinkin@nvidia.com>

Author: erickgalinkin

docs/user-guides/guardrails-library.md

@@ -1040,7 +1040,7 @@ To compute the perplexity of a string, the current implementation uses the `gpt2
 
 #### Model-based Jailbreak Detections
 
-There is currently one available model-based detection, using a random forest-based detector trained on [`snowflake/snowflake-arctic-embed-m-long`](https://huggingface.co/Snowflake/snowflake-arctic-embed-m-long) embeddings.
+There is currently one available model-based detection, using a random forest-based detector trained on [`Snowflake/snowflake-arctic-embed-m-long`](https://huggingface.co/Snowflake/snowflake-arctic-embed-m-long) embeddings.
 
 #### Setup
 

examples/configs/jailbreak_detection/config.yml

@@ -9,7 +9,7 @@ rails:
       server_endpoint: "http://localhost:1337/heuristics"
       lp_threshold: 89.79
       ps_ppl_threshold: 1845.65
-      embedding: "snowflake/snowflake-arctic-embed-m-long"
+      embedding: "Snowflake/snowflake-arctic-embed-m-long"
 
   input:
     flows:

examples/notebooks/safeguard_ai_virtual_assistant_notebook.ipynb

@@ -432,7 +432,7 @@
     "  config:\n",
     "    jailbreak_detection:\n",
     "      server_endpoint: \"\"\n",
-    "      embedding: \"snowflake/snowflake-arctic-embed-m-long\"\n",
+    "      embedding: \"Snowflake/snowflake-arctic-embed-m-long\"\n",
     "  input:\n",
     "    flows:\n",
     "      - content safety check input $model=llama-3.1-nemoguard-8b-content-safety\n",

nemoguardrails/library/jailbreak_detection/model_based/models.py

@@ -25,10 +25,10 @@ def __init__(self):
 
         self.device = "cuda:0" if torch.cuda.is_available() else "cpu"
         self.tokenizer = AutoTokenizer.from_pretrained(
-            "snowflake/snowflake-arctic-embed-m-long"
+            "Snowflake/snowflake-arctic-embed-m-long"
         )
         self.model = AutoModel.from_pretrained(
-            "snowflake/snowflake-arctic-embed-m-long",
+            "Snowflake/snowflake-arctic-embed-m-long",
             trust_remote_code=True,
             add_pooling_layer=False,
             safe_serialization=True,