From 9e4d0e733e4e6488a3a187bd34a750cc44b7ec4e Mon Sep 17 00:00:00 2001 From: leejet Date: Fri, 25 Sep 2026 16:28:07 +0800 Subject: [PATCH] fix: load safetensors index shards without recursion --- 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 2bd9261db..ac97cb7ed 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; } }