diff --git a/llama_stack/providers/remote/inference/ollama/ollama.py b/llama_stack/providers/remote/inference/ollama/ollama.py index 27bf0088e..70a091b77 100644 --- a/llama_stack/providers/remote/inference/ollama/ollama.py +++ b/llama_stack/providers/remote/inference/ollama/ollama.py @@ -44,10 +44,18 @@ model_aliases = [ "llama3.1:8b-instruct-fp16", CoreModelId.llama3_1_8b_instruct.value, ), + build_model_alias( + "llama3.1:8b", + CoreModelId.llama3_1_8b_instruct.value, + ), build_model_alias( "llama3.1:70b-instruct-fp16", CoreModelId.llama3_1_70b_instruct.value, ), + build_model_alias( + "llama3.1:70b", + CoreModelId.llama3_1_70b_instruct.value, + ), build_model_alias( "llama3.2:1b-instruct-fp16", CoreModelId.llama3_2_1b_instruct.value, @@ -56,6 +64,14 @@ model_aliases = [ "llama3.2:3b-instruct-fp16", CoreModelId.llama3_2_3b_instruct.value, ), + build_model_alias( + "llama3.2:1b", + CoreModelId.llama3_2_1b_instruct.value, + ), + build_model_alias( + "llama3.2:3b", + CoreModelId.llama3_2_3b_instruct.value, + ), build_model_alias( "llama-guard3:8b", CoreModelId.llama_guard_3_8b.value, @@ -68,6 +84,10 @@ model_aliases = [ "x/llama3.2-vision:11b-instruct-fp16", CoreModelId.llama3_2_11b_vision_instruct.value, ), + build_model_alias( + "llama3.2-vision", + CoreModelId.llama3_2_11b_vision_instruct.value, + ), ]