zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-08-21 20:46:46 +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}; return {arr};
} }
throw std::invalid_argument(
"[load_npy] Input must be a file-like object, string, or pathlib.Path");
} }
DictOrArray mlx_load_helper( DictOrArray mlx_load_helper(

5
python/src/load.h
View File

@ -22,7 +22,10 @@ void mlx_save_safetensor_helper(
py::dict d, py::dict d,
bool retain_graph = true); 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( void mlx_save_helper(
py::object file, py::object file,
array a, array a,