DarthAffe/stable-diffusion.cpp
1
0
Fork 0
mirror of https://github.com/leejet/stable-diffusion.cpp.git synced 2025-12-13 05:48:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

prevent errors when loading models without 'alphas_cumprod'

Browse Source
This commit is contained in:
leejet 2024-01-01 16:17:33 +08:00
parent 66be6e2e58
commit c429774fe9
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
model.cpp
View File

@ -1418,6 +1418,10 @@ bool ModelLoader::load_tensors(std::map<std::string, struct ggml_tensor*>& tenso
continue; continue;
} }
if (pair.first.find("alphas_cumprod") != std::string::npos) {
continue;
}
if (tensor_names_in_file.find(pair.first) == tensor_names_in_file.end()) { if (tensor_names_in_file.find(pair.first) == tensor_names_in_file.end()) {
LOG_ERROR("tensor '%s' not in model file", pair.first.c_str()); LOG_ERROR("tensor '%s' not in model file", pair.first.c_str());
some_tensor_not_init = true; some_tensor_not_init = true;