feat: add boogu image support (#1688)

This commit is contained in:
leejet
2026-06-22 00:36:17 +08:00
committed by GitHub
parent 2bd249c971
commit b12098f5d0
13 changed files with 1054 additions and 15 deletions

View File

@@ -485,6 +485,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("double_stream_layers.0.img_instruct_attn.processor.img_to_q.weight") != std::string::npos) {
return VERSION_BOOGU_IMAGE;
}
if (tensor_storage.name.find("model.diffusion_model.layers.0.adaLN_sa_ln.weight") != std::string::npos) {
return VERSION_ERNIE_IMAGE;
}