upstream-archive/stable-diffusion.cpp
1
0
Fork 0
mirror of https://github.com/leejet/stable-diffusion.cpp.git synced 2026-07-30 06:40:41 -05:00
Code Issues Packages Projects Releases Wiki Activity

feat: add ernie image support (#1427)

Browse Source
This commit is contained in:
leejet
2026-04-17 00:51:42 +08:00
committed by GitHub
parent c41c5ded7a
commit 5c243db9a8
14 changed files with 699 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/model.cpp
View File

@@ -1049,6 +1049,9 @@ SDVersion ModelLoader::get_sd_version() {
if (tensor_storage.name.find("model.diffusion_model.cap_embedder.0.weight") != std::string::npos) {
return VERSION_Z_IMAGE;
}
if (tensor_storage.name.find("model.diffusion_model.layers.0.adaLN_sa_ln.weight") != std::string::npos) {
return VERSION_ERNIE_IMAGE;
}
if (tensor_storage.name.find("model.diffusion_model.blocks.0.cross_attn.norm_k.weight") != std::string::npos) {
is_wan = true;
}