From 4c3cf7543da480843dacea19d82664c64cbad73a Mon Sep 17 00:00:00 2001 From: leejet Date: Fri, 25 Sep 2026 16:40:20 +0800 Subject: [PATCH] fix: load safetensors index shards without recursion (#2058) --- src/model_loader.cpp | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/src/model_loader.cpp b/src/model_loader.cpp index 2bd9261d..ac97cb7e 100644 --- a/src/model_loader.cpp +++ b/src/model_loader.cpp @@ -322,7 +322,13 @@ bool ModelLoader::init_from_safetensors_index_file(const std::string& file_path, } for (const std::string& shard_path : shard_paths) { - if (!parse_file(shard_path, prefix)) { + FileStamp stamp; + if (!read_file_stamp(shard_path, stamp)) { + return false; + } + parsed_dependencies_.push_back(stamp); + LOG_INFO("load %s using safetensors format", shard_path.c_str()); + if (!init_from_safetensors_file(shard_path, prefix)) { return false; } }