Added casing for inventory verbs

Author: Jack-Khuu

cli.py

@@ -49,8 +49,9 @@ def add_arguments_for_verb(parser, verb: str) -> None:
     # an explicit --checkpoint-dir, --checkpoint-path, or --tokenizer-path.
 
     if verb in INVENTORY_VERBS:
-       _configure_artifact_inventory_args(parser, verb)
-       return  
+        _configure_artifact_inventory_args(parser, verb)
+        _add_cli_metadata_args(parser)
+        return  
 
     parser.add_argument(
         "model",
@@ -200,12 +201,6 @@ def add_arguments_for_verb(parser, verb: str) -> None:
         choices=allowable_dtype_names(),
         help="Override the dtype of the model (default is the checkpoint dtype). Options: bf16, fp16, fp32, fast16, fast",
     )
-    parser.add_argument(
-        "-v",
-        "--verbose",
-        action="store_true",
-        help="Verbose output",
-    )
     parser.add_argument(
         "--quantize",
         type=str,
@@ -261,7 +256,17 @@ def add_arguments_for_verb(parser, verb: str) -> None:
         default=5000,
         help="Port for the web server in browser mode",
     )
+    _add_cli_metadata_args(parser)
+
 
+# Add CLI Args that are relevant to any subcommand execution
+def _add_cli_metadata_args(parser) -> None:
+    parser.add_argument(
+        "-v",
+        "--verbose",
+        action="store_true",
+        help="Verbose output",
+    )
 
 # Configure CLI Args specific to Model Artifact Management
 def _configure_artifact_inventory_args(parser, verb: str) -> None:
@@ -314,6 +319,7 @@ def _add_evaluation_args(parser) -> None:
     )
 
 
+# TODO: Refactor arg_init to be more modular
 def arg_init(args):
     if not (torch.__version__ > "2.3"):
         raise RuntimeError(
@@ -323,6 +329,11 @@ def arg_init(args):
     if sys.version_info.major != 3 or sys.version_info.minor < 10:
        raise RuntimeError("Please use Python 3.10 or later.")
 
+    # TODO: Don't initialize for Inventory management subcommands
+    # Remove when arg_init is refactored
+    if args.command in INVENTORY_VERBS:
+        return args
+
     if hasattr(args, "quantize") and Path(args.quantize).is_file():
         with open(args.quantize, "r") as f:
             args.quantize = json.loads(f.read())

download.py

@@ -125,13 +125,6 @@ def is_model_downloaded(model: str, models_dir: Path) -> bool:
 
 # Subcommand to list available models.
 def list_main(args) -> None:
-    # TODO It would be nice to have argparse validate this. However, we have
-    # model as an optional named parameter for all subcommands, so we'd
-    # probably need to move it to be registered per-command.
-    if args.model:
-        print("Usage: torchchat.py list")
-        return
-
     model_configs = load_model_configs()
 
     # Build the table in-memory so that we can align the text nicely.