fix: correct tae for models that use the flux2 vae (#1571)

This commit is contained in:
Wagner Bruna 2026-05-27 22:13:16 -03:00 committed by GitHub
parent 29ab511fc7
commit 0e4ee04488
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -584,7 +584,7 @@ public:
TAESD(bool decode_only = true, SDVersion version = VERSION_SD1)
: decode_only(decode_only) {
bool use_midblock_gn = false;
taef2 = sd_version_is_flux2(version);
taef2 = sd_version_uses_flux2_vae(version);
if (sd_version_is_dit(version)) {
z_channels = 16;