phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-03 01:03:59 +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]: def prompt_guard_download_info_map() -> Dict[str, LlamaDownloadInfo]:
return { return {
model.model_id: LlamaDownloadInfo( model.model_id: LlamaDownloadInfo(
folder=model.model_id, folder="Prompt-Guard" if model.model_id == "Prompt-Guard-86M" else model.model_id,
files=[ files=[
"model.safetensors", "model.safetensors",
"special_tokens_map.json", "special_tokens_map.json",