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

read existing configuration, save enums properly

Browse source
This commit is contained in:
Ashwin Bharambe 2024-08-02 13:55:29 -07:00
parent 2cf9915806
commit 3bc827cd5f
3 changed files with 63 additions and 26 deletions
Download patch file Download diff file Expand all files Collapse all files

10
llama_toolchain/cli/model/describe.py
View file

@ -7,21 +7,13 @@
import argparse
import json
from enum import Enum
from llama_models.sku_list import resolve_model
from termcolor import colored
from llama_toolchain.cli.subcommand import Subcommand
from llama_toolchain.cli.table import print_table
class EnumEncoder(json.JSONEncoder):
def default(self, obj):
if isinstance(obj, Enum):
return obj.value
return super().default(obj)
from llama_toolchain.utils import EnumEncoder
class ModelDescribe(Subcommand):