diff --git a/llama_stack/providers/remote/inference/groq/groq.py b/llama_stack/providers/remote/inference/groq/groq.py index 4e6cc2d6b..87c4d0d00 100644 --- a/llama_stack/providers/remote/inference/groq/groq.py +++ b/llama_stack/providers/remote/inference/groq/groq.py @@ -65,7 +65,7 @@ _MODEL_ALIASES = [ # TODO(aidand): Replace this with a stable model once Groq supports it build_model_alias( "llama-3.2-3b-preview", - CoreModelId.llama3_2_3b_instruct.value, + CoreModelId.llama3_2_3b.value, ), ] @@ -105,7 +105,7 @@ class GroqInferenceAdapter(Inference, ModelRegistryHelper, NeedsRequestProviderD model_id = self.get_provider_model_id(model_id) if model_id == "llama-3.2-3b-preview": warnings.warn( - "Groq only contains a preview version for llama-3.2-3b-instruct. " + "Groq only contains a preview version for llama-3.2-3b. " "Preview models aren't recommended for production use. " "They can be discontinued on short notice." )