From 19ce6bf009a80dbc5ae269532b944e3579764fbd Mon Sep 17 00:00:00 2001 From: Ashwin Bharambe Date: Wed, 2 Oct 2024 20:43:57 -0700 Subject: [PATCH] Don't validate prompt-guard anymore --- .../impls/meta_reference/safety/config.py | 14 -------------- 1 file changed, 14 deletions(-) diff --git a/llama_stack/providers/impls/meta_reference/safety/config.py b/llama_stack/providers/impls/meta_reference/safety/config.py index 734103412..36428078d 100644 --- a/llama_stack/providers/impls/meta_reference/safety/config.py +++ b/llama_stack/providers/impls/meta_reference/safety/config.py @@ -50,20 +50,6 @@ class LlamaGuardShieldConfig(BaseModel): class PromptGuardShieldConfig(BaseModel): model: str = "Prompt-Guard-86M" - @validator("model") - @classmethod - def validate_model(cls, model: str) -> str: - permitted_models = [ - m.descriptor() - for m in safety_models() - if m.core_model_id == CoreModelId.prompt_guard_86m - ] - if model not in permitted_models: - raise ValueError( - f"Invalid model: {model}. Must be one of {permitted_models}" - ) - return model - class SafetyConfig(BaseModel): llama_guard_shield: Optional[LlamaGuardShieldConfig] = None