DarthAffe/stable-diffusion.cpp
1
0
Fork 0
mirror of https://github.com/leejet/stable-diffusion.cpp.git synced 2026-06-09 15:56:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: use flux flow prediction for LTXAV (#1561)

Browse Source
This commit is contained in:
leejet 2026-05-26 00:23:39 +08:00 committed by GitHub
parent a397e03488
commit 202c6154a2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/stable-diffusion.cpp
View File

@ -1098,7 +1098,6 @@ public:
pred_type = EPS_PRED; pred_type = EPS_PRED;
} }
} else if (sd_version_is_sd3(version) || } else if (sd_version_is_sd3(version) ||
sd_version_is_ltxav(version) ||
sd_version_is_wan(version) || sd_version_is_wan(version) ||
sd_version_is_qwen_image(version) || sd_version_is_qwen_image(version) ||
version == VERSION_HIDREAM_O1 || version == VERSION_HIDREAM_O1 ||
@ -1106,16 +1105,16 @@ public:
sd_version_is_ernie_image(version) || sd_version_is_ernie_image(version) ||
sd_version_is_z_image(version)) { sd_version_is_z_image(version)) {
pred_type = FLOW_PRED; pred_type = FLOW_PRED;
if (sd_version_is_ltxav(version)) { if (sd_version_is_wan(version)) {
default_flow_shift = 2.37f;
} else if (sd_version_is_wan(version)) {
default_flow_shift = 5.f; default_flow_shift = 5.f;
} else if (sd_version_is_ernie_image(version)) { } else if (sd_version_is_ernie_image(version)) {
default_flow_shift = 4.f; default_flow_shift = 4.f;
} else { } else {
default_flow_shift = 3.f; default_flow_shift = 3.f;
} }
} else if (sd_version_is_flux(version) || sd_version_is_longcat(version)) { } else if (sd_version_is_flux(version) ||
sd_version_is_longcat(version) ||
sd_version_is_ltxav(version)) {
pred_type = FLUX_FLOW_PRED; pred_type = FLUX_FLOW_PRED;
default_flow_shift = 1.0f; // TODO: validate default_flow_shift = 1.0f; // TODO: validate
@ -1127,6 +1126,8 @@ public:
} }
if (sd_version_is_longcat(version)) { if (sd_version_is_longcat(version)) {
default_flow_shift = 3.0f; default_flow_shift = 3.0f;
} else if (sd_version_is_ltxav(version)) {
default_flow_shift = 2.37f;
} }
} else if (sd_version_is_flux2(version)) { } else if (sd_version_is_flux2(version)) {
pred_type = FLUX2_FLOW_PRED; pred_type = FLUX2_FLOW_PRED;