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

fix cli describe

Browse source
This commit is contained in:
Xi Yan 2024-09-24 14:41:13 -07:00
parent 00352bd251
commit c4534217c8
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
llama_stack/cli/model/describe.py
View file

@ -9,12 +9,12 @@ import json
from llama_models.sku_list import resolve_model from llama_models.sku_list import resolve_model
from termcolor import colored
from llama_stack.cli.subcommand import Subcommand from llama_stack.cli.subcommand import Subcommand
from llama_stack.cli.table import print_table from llama_stack.cli.table import print_table
from llama_stack.distribution.utils.serialize import EnumEncoder from llama_stack.distribution.utils.serialize import EnumEncoder
from termcolor import colored
class ModelDescribe(Subcommand): class ModelDescribe(Subcommand):
"""Show details about a model""" """Show details about a model"""
@ -52,7 +52,7 @@ class ModelDescribe(Subcommand):
colored(model.descriptor(), "white", attrs=["bold"]), colored(model.descriptor(), "white", attrs=["bold"]),
), ),
("HuggingFace ID", model.huggingface_repo or "<Not Available>"), ("HuggingFace ID", model.huggingface_repo or "<Not Available>"),
("Description", model.description_markdown), ("Description", model.description),
("Context Length", f"{model.max_seq_length // 1024}K tokens"), ("Context Length", f"{model.max_seq_length // 1024}K tokens"),
("Weights format", model.quantization_format.value), ("Weights format", model.quantization_format.value),
("Model params.json", json.dumps(model.arch_args, indent=4)), ("Model params.json", json.dumps(model.arch_args, indent=4)),