phoenix-oss/llama-stack
1
0
Fork 0
forked from phoenix-oss/llama-stack-mirror
Code Pull requests Releases Packages 2 Activity Actions

hide non-featured (older) models from model list command without show-all flag (#23)

Browse source
This commit is contained in:
Dalton Flanagan 2024-08-07 23:31:30 -04:00 • committed by GitHub
parent 7664d5701d
commit da4645a27a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
llama_toolchain/cli/model/list.py
View file

@ -27,7 +27,7 @@ class ModelList(Subcommand):
self.parser.set_defaults(func=self._run_model_list_cmd) self.parser.set_defaults(func=self._run_model_list_cmd)
def _add_arguments(self): def _add_arguments(self):
pass self.parser.add_argument('--show-all', action='store_true', help='Show all models (not just defaults)')
def _run_model_list_cmd(self, args: argparse.Namespace) -> None: def _run_model_list_cmd(self, args: argparse.Namespace) -> None:
headers = [ headers = [
@ -39,6 +39,9 @@ class ModelList(Subcommand):
rows = [] rows = []
for model in all_registered_models(): for model in all_registered_models():
if not args.show_all and not model.is_featured:
continue
req = model.hardware_requirements req = model.hardware_requirements
descriptor = model.descriptor() descriptor = model.descriptor()