mirror of
https://github.com/invoke-ai/InvokeAI.git
synced 2026-04-23 03:00:31 -04:00
* fix(model_manager): detect Flux VAE by latent space dimensions instead of filename VAE detection previously relied solely on filename pattern matching, which failed for Flux VAE files with generic names like "ae.safetensors". Now probes the model's decoder.conv_in weight shape to determine the latent space dimensions: - 16 channels -> Flux VAE - 4 channels -> SD/SDXL VAE (with filename fallback for SD1/SD2/SDXL distinction) * fix(model_manager): add latent space probing for Flux2 VAE detection Extend Flux2 VAE detection to also check for 32-dimensional latent space (decoder.conv_in with 32 input channels) in addition to BatchNorm layers. This provides more robust detection for Flux2 VAE files regardless of filename. * Chore Ruff format --------- Co-authored-by: Lincoln Stein <lincoln.stein@gmail.com>