zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-08-21 11:18:06 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

comment

Browse Source
This commit is contained in:
Awni Hannun 2023-12-23 06:42:41 -08:00
parent 1baff8c2b4
commit 51322b90e3
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
python/src/ops.cpp
View File

@ -2929,12 +2929,12 @@ void init_ops(py::module_& m) {
"load", "load",
&mlx_load_helper, &mlx_load_helper,
"file"_a, "file"_a,
"format"_a = none,
py::pos_only(), py::pos_only(),
"format"_a = none,
py::kw_only(), py::kw_only(),
"stream"_a = none, "stream"_a = none,
R"pbdoc( R"pbdoc(
load(file: str, format: Optional[str] = None, /, *, stream: Union[None, Stream, Device] = None) -> Union[array, Dict[str, array]] load(file: str, /, format: Optional[str] = None, *, stream: Union[None, Stream, Device] = None) -> Union[array, Dict[str, array]]
Load array(s) from a binary file in ``.npy``, ``.npz``, or ``.safetensors`` format. Load array(s) from a binary file in ``.npy``, ``.npz``, or ``.safetensors`` format.