sku -> family

This commit is contained in:
Ashwin Bharambe 2024-07-22 11:15:04 -07:00
parent f0e0903270
commit c38d638340

View file

@ -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",