Skip to content

Commit ab71f6d

Browse files
committed
support ovis, added taehv wan embed
1 parent 44ce1a8 commit ab71f6d

File tree

2 files changed

+5
-8
lines changed

2 files changed

+5
-8
lines changed

embd_res/taesd_w21.embd

10.8 MB
Binary file not shown.

otherarch/sdcpp/stable-diffusion.cpp

Lines changed: 5 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -285,9 +285,10 @@ class StableDiffusionGGML {
285285
bool isqwenimg = (tempver==VERSION_QWEN_IMAGE);
286286
bool iszimg = (tempver==VERSION_Z_IMAGE);
287287
bool isflux2 = (tempver==VERSION_FLUX2);
288+
bool is_ovis = (tempver==VERSION_OVIS_IMAGE);
288289

289290
//kcpp qol fallback: if qwen image, and they loaded the qwen2vl llm as t5 by mistake
290-
if((isqwenimg||iszimg||isflux2) && t5_path_fixed!="")
291+
if((isqwenimg||iszimg||isflux2||is_ovis) && t5_path_fixed!="")
291292
{
292293
if(clipl_path_fixed=="" && clipg_path_fixed=="")
293294
{
@@ -319,7 +320,7 @@ class StableDiffusionGGML {
319320
prefix = "cond_stage_model.transformer.";
320321
LOG_INFO("swap clip_vision from '%s'", clipl_path_fixed.c_str());
321322
}
322-
if(isqwenimg||iszimg||isflux2)
323+
if(isqwenimg||iszimg||isflux2||is_ovis)
323324
{
324325
prefix = "text_encoders.llm.";
325326
LOG_INFO("swap llm from '%s'", clipl_path_fixed.c_str());
@@ -408,19 +409,15 @@ class StableDiffusionGGML {
408409
{
409410
to_replace = "taesd_xl.embd";
410411
}
411-
else if(sd_version_is_flux(version)||sd_version_is_z_image(version))
412+
else if(sd_version_is_flux(version)||sd_version_is_z_image(version)||version == VERSION_OVIS_IMAGE)
412413
{
413414
to_replace = "taesd_f.embd";
414415
}
415416
else if(sd_version_is_sd3(version))
416417
{
417418
to_replace = "taesd_3.embd";
418419
}
419-
else if(version == VERSION_WAN2_2_TI2V)
420-
{
421-
to_replace = "taesd_w22.embd";
422-
}
423-
else if(sd_version_is_wan(version)||sd_version_is_qwen_image(version))
420+
else if((sd_version_is_wan(version) && version != VERSION_WAN2_2_TI2V)||sd_version_is_qwen_image(version))
424421
{
425422
to_replace = "taesd_w21.embd";
426423
}

0 commit comments

Comments
 (0)