From a042f41d036d0bb63d51e152e9f606ff908399b0 Mon Sep 17 00:00:00 2001 From: Dinesh Yeduguru Date: Mon, 18 Nov 2024 20:55:54 -0800 Subject: [PATCH] fix core model ids for ollama --- .../providers/remote/inference/ollama/ollama.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/llama_stack/providers/remote/inference/ollama/ollama.py b/llama_stack/providers/remote/inference/ollama/ollama.py index 70a091b77..5f783d982 100644 --- a/llama_stack/providers/remote/inference/ollama/ollama.py +++ b/llama_stack/providers/remote/inference/ollama/ollama.py @@ -46,7 +46,7 @@ model_aliases = [ ), build_model_alias( "llama3.1:8b", - CoreModelId.llama3_1_8b_instruct.value, + CoreModelId.llama3_1_8b.value, ), build_model_alias( "llama3.1:70b-instruct-fp16", @@ -54,7 +54,7 @@ model_aliases = [ ), build_model_alias( "llama3.1:70b", - CoreModelId.llama3_1_70b_instruct.value, + CoreModelId.llama3_1_70b.value, ), build_model_alias( "llama3.2:1b-instruct-fp16", @@ -66,11 +66,11 @@ model_aliases = [ ), build_model_alias( "llama3.2:1b", - CoreModelId.llama3_2_1b_instruct.value, + CoreModelId.llama3_2_1b.value, ), build_model_alias( "llama3.2:3b", - CoreModelId.llama3_2_3b_instruct.value, + CoreModelId.llama3_2_3b.value, ), build_model_alias( "llama-guard3:8b", @@ -85,8 +85,8 @@ model_aliases = [ CoreModelId.llama3_2_11b_vision_instruct.value, ), build_model_alias( - "llama3.2-vision", - CoreModelId.llama3_2_11b_vision_instruct.value, + "llama3.2-vision:11b", + CoreModelId.llama3_2_11b_vision.value, ), ]