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

Add a verify-download command to llama CLI

Browse source
This commit is contained in:
Ashwin Bharambe 2024-11-14 09:57:21 -08:00
parent 0713607b68
commit ba8d2369b8
4 changed files with 178 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_stack/cli/llama.py
View file

@ -9,6 +9,7 @@ import argparse
from .download import Download
from .model import ModelParser
from .stack import StackParser
from .verify_download import VerifyDownload
class LlamaCLIParser:
@ -27,9 +28,10 @@ class LlamaCLIParser:
subparsers = self.parser.add_subparsers(title="subcommands")
# Add sub-commands
Download.create(subparsers)
ModelParser.create(subparsers)
StackParser.create(subparsers)
Download.create(subparsers)
VerifyDownload.create(subparsers)
def parse_args(self) -> argparse.Namespace:
return self.parser.parse_args()