From 2795731434181e647179edb8d48c7f1e54deef2c Mon Sep 17 00:00:00 2001 From: Connor Hack Date: Wed, 20 Nov 2024 14:37:27 -0800 Subject: [PATCH] Update model name for mete-reference template --- .github/workflows/gha_workflow_llama_stack_tests.yml | 3 +++ llama_stack/templates/meta-reference-gpu/meta_reference.py | 2 +- 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/.github/workflows/gha_workflow_llama_stack_tests.yml b/.github/workflows/gha_workflow_llama_stack_tests.yml index 8f7a25ee4..ee3451d7b 100644 --- a/.github/workflows/gha_workflow_llama_stack_tests.yml +++ b/.github/workflows/gha_workflow_llama_stack_tests.yml @@ -1,6 +1,9 @@ name: "Run Llama-stack Tests" on: + push: + branches: + - 'main' pull_request_target: types: ["opened"] branches: diff --git a/llama_stack/templates/meta-reference-gpu/meta_reference.py b/llama_stack/templates/meta-reference-gpu/meta_reference.py index f254bc920..2408ef972 100644 --- a/llama_stack/templates/meta-reference-gpu/meta_reference.py +++ b/llama_stack/templates/meta-reference-gpu/meta_reference.py @@ -81,7 +81,7 @@ def get_distribution_template() -> DistributionTemplate: "Port for the Llama Stack distribution server", ), "INFERENCE_MODEL": ( - "meta-llama/Llama-3.2-3B-Instruct", + "meta-llama/Llama3.2-3B-Instruct", "Inference model loaded into the Meta Reference server", ), "INFERENCE_CHECKPOINT_DIR": (