From ccd90b2fa0c505220eb6b55fa5124655f4d19482 Mon Sep 17 00:00:00 2001 From: Jash Gulabrai Date: Wed, 16 Apr 2025 17:19:50 -0400 Subject: [PATCH] fix: Add llama-3.2-1b-instruct to NVIDIA fine-tuned model list --- llama_stack/providers/remote/post_training/nvidia/models.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/llama_stack/providers/remote/post_training/nvidia/models.py b/llama_stack/providers/remote/post_training/nvidia/models.py index 7c696ac20..1b31b4dbe 100644 --- a/llama_stack/providers/remote/post_training/nvidia/models.py +++ b/llama_stack/providers/remote/post_training/nvidia/models.py @@ -16,7 +16,11 @@ _MODEL_ENTRIES = [ build_hf_repo_model_entry( "meta/llama-3.1-8b-instruct", CoreModelId.llama3_1_8b_instruct.value, - ) + ), + build_hf_repo_model_entry( + "meta/llama-3.2-1b-instruct", + CoreModelId.llama3_2_1b_instruct.value, + ), ]