fix: do not check weights of open clip last layer

2023-09-03 21:10:08 +08:00 · 2023-09-03 21:10:08 +08:00 · 3ffffa6929
commit 3ffffa6929
parent 45842865ff
1 changed files with 3 additions and 0 deletions
--- a/stable-diffusion.cpp
+++ b/stable-diffusion.cpp
@ -3029,6 +3029,9 @@ class StableDiffusionGGML {
            }
            bool some_tensor_not_init = false;
            for (auto pair : tensors) {
+                if (pair.first.find("cond_stage_model.transformer.text_model.encoder.layers.23") != std::string::npos) {
+                    continue;
+                }
                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());
                    some_tensor_not_init = true;