mirror of
https://github.com/LostRuins/koboldcpp.git
synced 2026-05-09 02:50:39 +00:00
adapt kcpp changes for z-image
This commit is contained in:
parent
fcb86f22df
commit
fbdcc74ae6
1 changed files with 3 additions and 2 deletions
|
|
@ -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";
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue