From c38d638340531063097964f72f6d97d91dea74fb Mon Sep 17 00:00:00 2001
From: Ashwin Bharambe <ashwin@meta.com>
Date: Mon, 22 Jul 2024 11:15:04 -0700
Subject: [PATCH] sku -> family

---
 llama_toolchain/cli/model/template.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/llama_toolchain/cli/model/template.py b/llama_toolchain/cli/model/template.py
index bdfe04bf7..90f109184 100644
--- a/llama_toolchain/cli/model/template.py
+++ b/llama_toolchain/cli/model/template.py
@@ -27,9 +27,11 @@ class ModelTemplate(Subcommand):
 
     def _add_arguments(self):
         self.parser.add_argument(
-            "sku",
+            "-m",
+            "--model-family",
             type=str,
-            help="Model SKU",
+            default="llama3_1",
+            help="Model Family (llama3_1, llama3_X, etc.)",
         )
         self.parser.add_argument(
             "--template",