Update llama_stack/providers/remote/inference/groq/groq.py

This commit is contained in:
Aidan Do 2024-12-14 16:36:16 +11:00 committed by GitHub
parent aa9a9f18be
commit b78b9ed4a1
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -104,7 +104,7 @@ class GroqInferenceAdapter(Inference, ModelRegistryHelper):
) -> Union[
ChatCompletionResponse, AsyncIterator[ChatCompletionResponseStreamChunk]
]:
model_id = self.get_provider_model_id(model_id)
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. "