Align CLI args and some smaller fixes (#167)

* Add `.DS_Store` files to `.gitignore`

* Fix variable naming of `config` in `mixtral/convert.py`

* Align CLI args and minor fixes

* standardize

* one more

---------

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

.gitignore

@@ -127,5 +127,10 @@ dmypy.json
 
 # Pyre type checker
 .pyre/
+
+# IDE files
 .idea/
 .vscode/
+
+# .DS_Store files
+.DS_Store

llms/llama/convert.py

@@ -60,7 +60,7 @@ def llama(model_path):
 def tiny_llama(model_path):
     try:
         import transformers
-    except ImportError as e:
+    except ImportError:
         print("The transformers package must be installed for this model conversion:")
         print("pip install transformers")
         exit(0)

llms/llama/llama.py

@@ -5,7 +5,7 @@ import json
 import time
 from dataclasses import dataclass
 from pathlib import Path
-from typing import List, Optional, Tuple
+from typing import Optional, Tuple
 
 import mlx.core as mx
 import mlx.nn as nn
@@ -218,7 +218,6 @@ def toc(msg, start):
 
 
 def generate(args):
-
     input("Press enter to start generation")
     print("------")
     print(args.prompt)
@@ -347,7 +346,7 @@ if __name__ == "__main__":
     parser = argparse.ArgumentParser(description="Llama inference script")
     parser.add_argument(
         "--model-path",
-        help="Path to the model directory containing the MLX weights",
+        help="Path to the model weights and tokenizer",
         default="mlx_model",
     )
     parser.add_argument(

llms/mixtral/mixtral.py

@@ -9,7 +9,6 @@ from typing import List, Optional, Tuple
 
 import mlx.core as mx
 import mlx.nn as nn
-import numpy as np
 from mlx.utils import tree_map, tree_unflatten
 from sentencepiece import SentencePieceProcessor