DarthAffe/stable-diffusion.cpp
1
0
Fork 0
mirror of https://github.com/leejet/stable-diffusion.cpp.git synced 2026-03-23 17:58:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: correct encoder channels for flux2 (#1346)

Browse Source
This commit is contained in:
Daniele 2026-03-16 15:16:43 +01:00 committed by GitHub
parent 862a6586cb
commit 997bb11fb6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/auto_encoder_kl.hpp
View File

@ -613,6 +613,9 @@ public:
int get_encoder_output_channels() { int get_encoder_output_channels() {
int factor = dd_config.double_z ? 2 : 1; int factor = dd_config.double_z ? 2 : 1;
if (sd_version_is_flux2(version)) {
return dd_config.z_channels * 4;
}
return dd_config.z_channels * factor; return dd_config.z_channels * factor;
} }
}; };
@ -927,4 +930,4 @@ struct AutoEncoderKL : public VAE {
}; };
}; };
#endif // __AUTO_ENCODER_KL_HPP__ #endif // __AUTO_ENCODER_KL_HPP__