diff --git a/llama_stack/providers/inline/inference/meta_reference/config.py b/llama_stack/providers/inline/inference/meta_reference/config.py index 9556b026a..961548f9c 100644 --- a/llama_stack/providers/inline/inference/meta_reference/config.py +++ b/llama_stack/providers/inline/inference/meta_reference/config.py @@ -39,7 +39,7 @@ class MetaReferenceInferenceConfig(BaseModel): def validate_model(cls, model: str) -> str: permitted_models = supported_inference_models() descriptors = [m.descriptor() for m in permitted_models] - repos = [m.huggingface_repo for m in permitted_models] + repos = [m.huggingface_repo for m in permitted_models if m.huggingface_repo is not None] if model not in (descriptors + repos): model_list = "\n\t".join(repos) raise ValueError(f"Unknown model: `{model}`. Choose from [\n\t{model_list}\n]") diff --git a/pyproject.toml b/pyproject.toml index 4a37f8c56..bab494be2 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -248,7 +248,6 @@ exclude = [ "^llama_stack/providers/inline/agents/meta_reference/agents\\.py$", "^llama_stack/providers/inline/datasetio/localfs/", "^llama_stack/providers/inline/eval/meta_reference/eval\\.py$", - "^llama_stack/providers/inline/inference/meta_reference/config\\.py$", "^llama_stack/providers/inline/inference/meta_reference/inference\\.py$", "^llama_stack/models/llama/llama3/generation\\.py$", "^llama_stack/models/llama/llama3/multimodal/model\\.py$",