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

more cleanup

Browse Source
This commit is contained in:
dc-dc-dc 2023-12-19 14:22:09 -05:00
parent d3b1bea251
commit c043d27682
2 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mlx/ops.h
View File

@ -1057,7 +1057,7 @@ array dequantize(
int bits = 4, int bits = 4,
StreamOrDevice s = {}); StreamOrDevice s = {});
/** Load array from .safetensor file format */ /** Load array map from .safetensor file format */
std::unordered_map<std::string, array> load_safetensor( std::unordered_map<std::string, array> load_safetensor(
std::shared_ptr<io::Reader> in_stream, std::shared_ptr<io::Reader> in_stream,
StreamOrDevice s = {}); StreamOrDevice s = {});

1
mlx/safetensor.cpp
View File

@ -85,7 +85,6 @@ std::unordered_map<std::string, array> load_safetensor(
char rawJson[jsonHeaderLength]; char rawJson[jsonHeaderLength];
in_stream->read(rawJson, jsonHeaderLength); in_stream->read(rawJson, jsonHeaderLength);
auto metadata = json::parse(rawJson, rawJson + jsonHeaderLength); auto metadata = json::parse(rawJson, rawJson + jsonHeaderLength);
// auto metadata = io::jsonDeserialize(json, jsonHeaderLength);
// Should always be an object on the top-level // Should always be an object on the top-level
if (!metadata.is_object()) { if (!metadata.is_object()) {
throw std::runtime_error( throw std::runtime_error(