update toolchain to work with updated imports from llama_models

This commit is contained in:
Ashwin Bharambe 2024-07-30 17:52:57 -07:00
parent 23014ea4d1
commit 1bc81eae7b
3 changed files with 6 additions and 14 deletions

View file

@ -9,7 +9,7 @@ import json
from enum import Enum
from llama_models.llama3_1.api.sku_list import llama3_1_model_list
from llama_models.sku_list import llama3_1_model_list
from termcolor import colored
@ -43,6 +43,7 @@ class ModelDescribe(Subcommand):
"-m",
"--model-id",
type=str,
required=True,
)
def _run_model_describe_cmd(self, args: argparse.Namespace) -> None: