fix cli (ml-explore#827)

* fix cli

* nit

* fix typo in subpackages

---------

Co-authored-by: Awni Hannun <awni@apple.com>

Author: solarpunkin

mlx_lm/cli.py

@@ -16,29 +16,37 @@ def main():
         "lora",
         "manage",
         "perplexity",
-        "quant.awq",
-        "quant.dwq",
-        "quant.dynamic_quant",
-        "quant.gptq",
+        "awq",
+        "dwq",
+        "dynamic_quant",
+        "gptq",
         "server",
         "upload",
     )
+    subpackages = {
+        "awq": "quant",
+        "dwq": "quant",
+        "dynamic_quant": "quant",
+        "gptq": "quant",
+    }
     if len(sys.argv) < 2:
         raise ValueError(f"CLI requires a subcommand in {subcommands}")
     subcommand = sys.argv.pop(1)
     if subcommand in subcommands:
+        if subpackage := subpackages.get(subcommand):
+            subcommand = f"{subpackage}.{subcommand}"
         submodule = importlib.import_module(f"mlx_lm.{subcommand}")
         submodule.main()
     elif subcommand == "--version":
         from mlx_lm import __version__
 
         print(__version__)
     elif subcommand in ("-h", "--help"):
-        print(f"The supported subcommand are {subcommands}")
+        print(f"The supported subcommands are {subcommands}")
         print()
         print(
             "For help on an individual subcommand, pass --help "
-            "to the subommand. For example: mlx_lm.generate --help"
+            "to the subcommand. For example: mlx_lm.generate --help"
         )
     else:
         raise ValueError(f"CLI requires a subcommand in {subcommands}")