Moved model inv check earlier

Author: Jack-Khuu

cli.py

@@ -319,7 +319,6 @@ 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(
@@ -329,11 +328,6 @@ 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())

torchchat.py

@@ -12,6 +12,7 @@
 from cli import (
     add_arguments_for_verb,
     KNOWN_VERBS,
+    INVENTORY_VERBS,
     arg_init,
     check_args,
 )
@@ -49,7 +50,11 @@
 
     # Now parse the arguments
     args = parser.parse_args()
-    args = arg_init(args)
+
+    # Don't initialize for Inventory management subcommands
+    # TODO: Remove when arg_init is refactored
+    if args.command not in INVENTORY_VERBS:
+        args = arg_init(args)
     logging.basicConfig(
         format="%(message)s", level=logging.DEBUG if args.verbose else logging.INFO
     )