phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-06-28 19:04:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

model template --template -> model template --name

Browse source
This commit is contained in:
Ashwin Bharambe 2024-07-29 18:21:05 -07:00
parent 45b8a7ffcd
commit 59574924de
2 changed files with 6 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

6
docs/cli_reference.md
View file

@ -96,13 +96,13 @@ tool-message-success
user-message-default 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|> <|start_header_id|>ipython<|end_header_id|>

6
llama_toolchain/cli/model/template.py
View file

@ -46,15 +46,15 @@ class ModelTemplate(Subcommand):
help="Model Family (llama3_1, llama3_X, etc.)", help="Model Family (llama3_1, llama3_X, etc.)",
) )
self.parser.add_argument( self.parser.add_argument(
"--template", "--name",
type=str, type=str,
help="Usecase template name (system_message, user_message, assistant_message, tool_message)...", help="Usecase template name (system_message, user_message, assistant_message, tool_message)...",
required=False, required=False,
) )
def _run_model_template_cmd(self, args: argparse.Namespace) -> None: def _run_model_template_cmd(self, args: argparse.Namespace) -> None:
if args.template: if args.name:
template, tokens_info = render_jinja_template(args.template) template, tokens_info = render_jinja_template(args.name)
rendered = "" rendered = ""
for tok, is_special in tokens_info: for tok, is_special in tokens_info:
if is_special: if is_special: