phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-04 20:14:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

get ollama working

Browse source
This commit is contained in:
Hardik Shah 2024-08-07 17:52:49 -07:00
parent ea50086190
commit 171a178783
9 changed files with 151 additions and 375 deletions
Download patch file Download diff file Expand all files Collapse all files

6
llama_toolchain/safety/meta_reference/config.py
View file

@ -10,14 +10,14 @@ from pydantic import BaseModel
class LlamaGuardShieldConfig(BaseModel):
model: str
excluded_categories: List[str]
model: str = "Llama-Guard-3-8B"
excluded_categories: List[str] = []
disable_input_check: bool = False
disable_output_check: bool = False
class PromptGuardShieldConfig(BaseModel):
model: str
model: str = "Prompt-Guard-86M"
class SafetyConfig(BaseModel):