DarthAffe/stable-diffusion.cpp
1
0 Fork 0
mirror of https://github.com/leejet/stable-diffusion.cpp.git synced 2026-11-11 19:36:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

fixed soem bugs; now photomaker model weights can be loaded into their tensor buffers

Browse Source
This commit is contained in:
bssrdf committed 2024-02-04 17:56:18 -05:00
1 parent cbfa702fad
commit 78651f2535
4 files changed
+36 -10
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

clip.hpp
+3 -2
View File
@@ -863,7 +863,7 @@ struct CLIPVisionModel {
return mem_size;
}
void map_by_name(std::map<std::string, struct ggml_tensor*>& tensors, const std::string prefix) {
void map_by_name(std::map<std::string, struct ggml_tensor*>& tensors, const std::string prefix, const std::string stripped) {
tensors[prefix + "post_layernorm.bias"] = post_ln_b;
tensors[prefix + "post_layernorm.weight"] = post_ln_w;
tensors[prefix + "pre_layrnorm.bias"] = pre_ln_b;
@@ -876,7 +876,8 @@ struct CLIPVisionModel {
std::string name = prefix + "encoder.layers." + std::to_string(i) + ".";
resblocks[i].map_by_name(tensors, prefix + "encoder.layers." + std::to_string(i) + ".");
}
tensors[prefix + "visual_projection"] = visual_projection;
// std::string stripped = prefix.replace(prefix.find("vision_model."), sizeof("vision_model.")-1, "");
tensors[stripped + "visual_projection.weight"] = visual_projection;
}