Update bindings/python/py_src/safetensors/mlx.py

Co-authored-by: Pedro Cuenca <pedro@huggingface.co>
huggingface · Jan 5, 2024 · 4bb3e10 · 4bb3e10
1 parent 2a0fbd7
commit 4bb3e10
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bindings/python/py_src/safetensors/mlx.py b/bindings/python/py_src/safetensors/mlx.py
@@ -80,7 +80,7 @@ def load(data: bytes) -> Dict[str, mx.array]:
             The content of a safetensors file
 
     Returns:
-        `Dict[str, mx.array]`: dictionary that contains name as key, value as `mx.array` on cpu
+        `Dict[str, mx.array]`: dictionary that contains name as key, value as `mx.array`
 
     Example: