phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-07-29 15:23:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge remote-tracking branch 'origin/main' into distros

Browse source
This commit is contained in:
Ashwin Bharambe 2024-08-07 21:55:29 -07:00
commit e33f402046
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()