From ef2b686ff418b0242f305d9d6d94c778314b2971 Mon Sep 17 00:00:00 2001 From: Ashwin Bharambe Date: Tue, 29 Apr 2025 10:03:37 -0700 Subject: [PATCH] Update safety_models.py --- llama_stack/cli/model/safety_models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llama_stack/cli/model/safety_models.py b/llama_stack/cli/model/safety_models.py index d3147c441..aaee1adcb 100644 --- a/llama_stack/cli/model/safety_models.py +++ b/llama_stack/cli/model/safety_models.py @@ -51,7 +51,7 @@ def prompt_guard_model_sku_map() -> Dict[str, Any]: def prompt_guard_download_info_map() -> Dict[str, LlamaDownloadInfo]: return { model.model_id: LlamaDownloadInfo( - folder=model.model_id, + folder="Prompt-Guard" if model.model_id == "Prompt-Guard-86M" else model.model_id, files=[ "model.safetensors", "special_tokens_map.json",