phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-07-29 07:14:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

sku -> family

Browse source
This commit is contained in:
Ashwin Bharambe 2024-07-22 11:15:04 -07:00
parent f0e0903270
commit c38d638340
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -27,9 +27,11 @@ class ModelTemplate(Subcommand):
def _add_arguments(self): def _add_arguments(self):
self.parser.add_argument( self.parser.add_argument(
"sku", "-m",
"--model-family",
type=str, type=str,
help="Model SKU", default="llama3_1",
help="Model Family (llama3_1, llama3_X, etc.)",
) )
self.parser.add_argument( self.parser.add_argument(
"--template", "--template",