phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-06-28 19:04:19 +00:00
Code Issues Projects Releases Packages Wiki 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)
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:
headers = [
@ -39,6 +39,9 @@ class ModelList(Subcommand):
rows = []
for model in all_registered_models():
if not args.show_all and not model.is_featured:
continue
req = model.hardware_requirements
descriptor = model.descriptor()