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

small fixes

Browse Source
This commit is contained in:
dc-dc-dc 2023-12-22 01:25:42 -05:00
parent fdf9d99f0f
commit 13fd7bd721
2 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
python/src/load.cpp
View File

@ -232,6 +232,8 @@ array mlx_load_npy_helper(py::object file, StreamOrDevice s) {
}
return {arr};
}
throw std::invalid_argument(
"[load_npy] Input must be a file-like object, string, or pathlib.Path");
}
DictOrArray mlx_load_helper(

5
python/src/load.h
View File

@ -22,7 +22,10 @@ void mlx_save_safetensor_helper(
py::dict d,
bool retain_graph = true);
DictOrArray mlx_load_helper(py::object file, StreamOrDevice s);
DictOrArray mlx_load_helper(
py::object file,
std::optional<std::string> format,
StreamOrDevice s);
void mlx_save_helper(
py::object file,
array a,