phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-06-27 18:50:41 +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
```
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|>

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

@ -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: