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

ignore __metadata__

Browse Source
This commit is contained in:
dc-dc-dc 2023-12-18 17:18:55 -05:00
parent f4d876d35f
commit 801c1ad3a4
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mlx/safetensor.cpp
View File

@ -288,6 +288,10 @@ std::unordered_map<std::string, array> load_safetensor(
// Load the arrays using metadata // Load the arrays using metadata
std::unordered_map<std::string, array> res; std::unordered_map<std::string, array> res;
for (auto& [key, obj] : *metadata.getObject()) { for (auto& [key, obj] : *metadata.getObject()) {
if (key == "__metadata__") {
// ignore metadata for now
continue;
}
std::string dtype = obj->getObject()->at("dtype")->getString(); std::string dtype = obj->getObject()->at("dtype")->getString();
auto shape = obj->getObject()->at("shape")->getList(); auto shape = obj->getObject()->at("shape")->getList();
std::vector<int> shape_vec; std::vector<int> shape_vec;