From 47c23bfc6ea18da91bda618bb5e5dbe8e8e5228b Mon Sep 17 00:00:00 2001 From: varunfb Date: Tue, 10 Dec 2024 19:51:11 -0800 Subject: [PATCH] Added support for llama 3.3 model --- llama_stack/providers/utils/inference/__init__.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/llama_stack/providers/utils/inference/__init__.py b/llama_stack/providers/utils/inference/__init__.py index d204f98a4..553d02418 100644 --- a/llama_stack/providers/utils/inference/__init__.py +++ b/llama_stack/providers/utils/inference/__init__.py @@ -27,7 +27,8 @@ def supported_inference_models() -> List[Model]: m for m in all_registered_models() if ( - m.model_family in {ModelFamily.llama3_1, ModelFamily.llama3_2} + m.model_family + in {ModelFamily.llama3_1, ModelFamily.llama3_2, ModelFamily.llama3_3} or is_supported_safety_model(m) ) ]