update setup.py

ToluClassics · Apr 19, 2024 · 68e0523 · 68e0523
1 parent 1b7d4a2
commit 68e0523
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -45,6 +45,8 @@ A list of the available models can be found in the `mlx_transformers.models` mod
     config = BertConfig.from_pretrained("bert-base-uncased")
     model = MLXBertModel(config)
 
+    model.load_weights("bert-base-uncased.npz", strict=True)
+
     sample_input = "Hello, world!"
     inputs = tokenizer(sample_input, return_tensors="np")
     outputs = model(**inputs)

diff --git a/setup.py b/setup.py
@@ -12,7 +12,7 @@
 
 setuptools.setup(
     name="mlx-transformers",
-    version="0.0.1",
+    version="0.1.0",
     author="Ogundepo Odunayo",
     author_email="[email protected]",
     description=description,