From fbdcc74ae610a4cb7f85dd0c58284bfd49aeab30 Mon Sep 17 00:00:00 2001 From: Wagner Bruna Date: Mon, 1 Dec 2025 19:06:13 -0300 Subject: [PATCH] adapt kcpp changes for z-image --- otherarch/sdcpp/stable-diffusion.cpp | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/otherarch/sdcpp/stable-diffusion.cpp b/otherarch/sdcpp/stable-diffusion.cpp index 9bc8a8c43..cf0f78e8d 100644 --- a/otherarch/sdcpp/stable-diffusion.cpp +++ b/otherarch/sdcpp/stable-diffusion.cpp @@ -266,6 +266,7 @@ public: int tempver = model_loader.get_sd_version(); bool iswan = (tempver==VERSION_WAN2 || tempver==VERSION_WAN2_2_I2V || tempver==VERSION_WAN2_2_TI2V); bool isqwenimg = (tempver==VERSION_QWEN_IMAGE); + bool iszimg = (tempver==VERSION_Z_IMAGE); //kcpp qol fallback: if qwen image, and they loaded the qwen2vl llm as t5 by mistake if(isqwenimg && t5_path_fixed!="") @@ -300,7 +301,7 @@ public: prefix = "cond_stage_model.transformer."; LOG_INFO("swap clip_vision from '%s'", clipl_path_fixed.c_str()); } - if(isqwenimg) + if(isqwenimg||iszimg) { prefix = "text_encoders.llm."; LOG_INFO("swap llm from '%s'", clipl_path_fixed.c_str()); @@ -413,7 +414,7 @@ public: { to_replace = "taesd_xl.embd"; } - else if(sd_version_is_flux(version)) + else if(sd_version_is_flux(version)||sd_version_is_z_image(version)) { to_replace = "taesd_f.embd"; }