Add llama model download alias for llama download

llama_toolchain/cli/model/download.py

@@ -0,0 +1,24 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the terms described in the LICENSE file in
+# the root directory of this source tree.
+
+import argparse
+
+from llama_toolchain.cli.subcommand import Subcommand
+
+
+class ModelDownload(Subcommand):
+    def __init__(self, subparsers: argparse._SubParsersAction):
+        super().__init__()
+        self.parser = subparsers.add_parser(
+            "download",
+            prog="llama model download",
+            description="Download a model from llama.meta.comf or HuggingFace hub",
+            formatter_class=argparse.RawTextHelpFormatter,
+        )
+
+        from llama_toolchain.cli.download import setup_download_parser
+
+        setup_download_parser(self.parser)

llama_toolchain/cli/model/model.py

@@ -5,9 +5,9 @@
 # the root directory of this source tree.
 
 import argparse
-import textwrap
 
 from llama_toolchain.cli.model.describe import ModelDescribe
+from llama_toolchain.cli.model.download import ModelDownload
 from llama_toolchain.cli.model.list import ModelList
 from llama_toolchain.cli.model.template import ModelTemplate
 
@@ -22,18 +22,13 @@ class ModelParser(Subcommand):
         self.parser = subparsers.add_parser(
             "model",
             prog="llama model",
-            description="Describe llama model interfaces",
-            epilog=textwrap.dedent(
-                """
-                Example:
-                    llama model <subcommand> <options>
-                """
-            ),
+            description="Work with llama models",
         )
 
         subparsers = self.parser.add_subparsers(title="model_subcommands")
 
-        # Add sub-commandsa
-        ModelTemplate.create(subparsers)
+        # Add sub-commands
+        ModelDownload.create(subparsers)
         ModelList.create(subparsers)
+        ModelTemplate.create(subparsers)
         ModelDescribe.create(subparsers)