adapt kcpp changes for z-image

This commit is contained in:
Wagner Bruna 2025-12-01 19:06:13 -03:00
parent fcb86f22df
commit fbdcc74ae6

View file

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