DarthAffe/stable-diffusion.cpp
1
0
Fork 0
mirror of https://github.com/leejet/stable-diffusion.cpp.git synced 2026-09-25 12:40:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: load safetensors index shards without recursion (#2058)

Browse Source
This commit is contained in:
leejet 2026-09-25 16:40:20 +08:00 committed by GitHub
parent 510bccf330
commit 4c3cf7543d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/model_loader.cpp
View File

@ -322,7 +322,13 @@ bool ModelLoader::init_from_safetensors_index_file(const std::string& file_path,
} }
for (const std::string& shard_path : shard_paths) { 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; return false;
} }
} }