phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-02 08:44:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update safety_models.py

Browse source
This commit is contained in:
Ashwin Bharambe 2025-04-29 10:03:37 -07:00 • committed by GitHub
parent 38b580db02
commit ef2b686ff4
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/cli/model/safety_models.py
View file

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