phoenix-oss/llama-stack
1
0
Fork 0
forked from phoenix-oss/llama-stack-mirror
Code Pull requests Releases Packages 2 Activity Actions

Add special case for prompt guard

Browse source
This commit is contained in:
Ashwin Bharambe 2024-10-02 08:38:23 -07:00
parent a80b707ff8
commit cc5029a716
4 changed files with 76 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_stack/cli/model/list.py
View file

@ -34,6 +34,8 @@ class ModelList(Subcommand):
)
def _run_model_list_cmd(self, args: argparse.Namespace) -> None:
from .safety_models import prompt_guard_model_sku
headers = [
"Model Descriptor",
"Hugging Face Repo",
@ -41,7 +43,7 @@ class ModelList(Subcommand):
]
rows = []
for model in all_registered_models():
for model in all_registered_models() + [prompt_guard_model_sku()]:
if not args.show_all and not model.is_featured:
continue