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"; }