mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-06-28 02:53:30 +00:00
feat: add api.llama provider, llama-guard-4 model (#2058)
This PR adds a llama-stack inference provider for `api.llama.com`, as well as adds entries for Llama-Guard-4 and updated Prompt-Guard models.
This commit is contained in:
parent
934446ddb4
commit
4d0bfbf984
21 changed files with 1526 additions and 47 deletions
|
@ -792,6 +792,13 @@ def llama3_3_instruct_models() -> List[Model]:
|
|||
@lru_cache
|
||||
def safety_models() -> List[Model]:
|
||||
return [
|
||||
Model(
|
||||
core_model_id=CoreModelId.llama_guard_4_12b,
|
||||
description="Llama Guard v4 12b system safety model",
|
||||
huggingface_repo="meta-llama/Llama-Guard-4-12B",
|
||||
arch_args={},
|
||||
pth_file_count=1,
|
||||
),
|
||||
Model(
|
||||
core_model_id=CoreModelId.llama_guard_3_11b_vision,
|
||||
description="Llama Guard v3 11b vision system safety model",
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue