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

feat: add krea2 support (#1705)

Browse Source
This commit is contained in:
leejet
2026-06-25 00:25:50 +08:00
committed by GitHub
parent 2938272d82
commit 8caa3f908a
10 changed files with 803 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/model_loader.cpp
View File

@@ -453,6 +453,10 @@ SDVersion ModelLoader::get_sd_version() {
if (tensor_storage.name.find("embed_image_indicator.weight") != std::string::npos) {
return VERSION_IDEOGRAM4;
}
if (tensor_storage.name.find("model.diffusion_model.txtfusion.projector.weight") != std::string::npos ||
tensor_storage.name.find("model.diffusion_model.text_fusion.projector.weight") != std::string::npos) {
return VERSION_KREA2;
}
if (tensor_storage.name.find("model.diffusion_model.nerf_final_layer_conv.") != std::string::npos) {
return VERSION_CHROMA_RADIANCE;
}