mirror of
https://github.com/invoke-ai/InvokeAI.git
synced 2026-04-23 03:00:31 -04:00
Infer encoder and decoder block_out_channels independently from the state dict and rebuild the decoder submodule when its channel widths differ from the encoder, so the asymmetric full_encoder_small_decoder checkpoint from black-forest-labs/FLUX.2-small-decoder loads correctly. Co-authored-by: Lincoln Stein <lincoln.stein@gmail.com>