From 33117e301203b9c5649e598a73e92174b3462616 Mon Sep 17 00:00:00 2001 From: Sajikumar JS Date: Thu, 10 Apr 2025 10:17:43 +0530 Subject: [PATCH] Updated CoreModelId to get from sku_types --- llama_stack/providers/remote/inference/watsonx/models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llama_stack/providers/remote/inference/watsonx/models.py b/llama_stack/providers/remote/inference/watsonx/models.py index f700f4096..d54766698 100644 --- a/llama_stack/providers/remote/inference/watsonx/models.py +++ b/llama_stack/providers/remote/inference/watsonx/models.py @@ -4,7 +4,7 @@ # This source code is licensed under the terms described in the LICENSE file in # the root directory of this source tree. -from llama_stack.models.llama.datatypes import CoreModelId +from llama_stack.models.llama.sku_types import CoreModelId from llama_stack.providers.utils.inference.model_registry import build_hf_repo_model_entry MODEL_ENTRIES = [