From 59574924de341127555e15e492b447e22bd19dd8 Mon Sep 17 00:00:00 2001
From: Ashwin Bharambe <ashwin@meta.com>
Date: Mon, 29 Jul 2024 18:21:05 -0700
Subject: [PATCH] model template --template -> model template --name

---
 docs/cli_reference.md                 | 6 +++---
 llama_toolchain/cli/model/template.py | 6 +++---
 2 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/docs/cli_reference.md b/docs/cli_reference.md
index 404d13c19..8e9709c80 100644
--- a/docs/cli_reference.md
+++ b/docs/cli_reference.md
@@ -96,13 +96,13 @@ tool-message-success
 user-message-default
 ```
 
-And fetch an example by passing it to `--template`:
+And fetch an example by passing it to `--name`:
 
 ```
-llama model template --template tool-message-success
+llama model template --name tool-message-success
 
 
-llama model template --template tool-message-success
+llama model template --name tool-message-success
 <|start_header_id|>ipython<|end_header_id|>
 
 
diff --git a/llama_toolchain/cli/model/template.py b/llama_toolchain/cli/model/template.py
index 911456a46..1fdeab3e7 100644
--- a/llama_toolchain/cli/model/template.py
+++ b/llama_toolchain/cli/model/template.py
@@ -46,15 +46,15 @@ class ModelTemplate(Subcommand):
             help="Model Family (llama3_1, llama3_X, etc.)",
         )
         self.parser.add_argument(
-            "--template",
+            "--name",
             type=str,
             help="Usecase template name (system_message, user_message, assistant_message, tool_message)...",
             required=False,
         )
 
     def _run_model_template_cmd(self, args: argparse.Namespace) -> None:
-        if args.template:
-            template, tokens_info = render_jinja_template(args.template)
+        if args.name:
+            template, tokens_info = render_jinja_template(args.name)
             rendered = ""
             for tok, is_special in tokens_info:
                 if is_special: