flux2 vae swap setting

Author: mcmonkey4eva

docs/Model Support.md

@@ -331,6 +331,7 @@ These steps are not friendly to beginners (if Sana gains popularity, likely more
     - Or GGUF version here <https://huggingface.co/orabazes/FLUX.2-dev-GGUF/tree/main>
     - Goes in `diffusion_models` folder
 - The VAE is a brand new 16x16 downsample VAE with 128 channels. It will be autodownloaded.
+    - You can swap it for this one <https://huggingface.co/CabalResearch/Flux2VAE-Anime-Decoder-Tune/blob/main/Flux2Anime%20VAE%20DecB1.safetensors> which is a finetune to reduce detail artifacting.
 - The Text Encoder is 24B Mistral Small 3.2 (2506). It will be autodownloaded.
     - This would make sense to replace with a GGUF. (Pending gguf download link)
 - **Parameters:**

src/BuiltinExtensions/ComfyUIBackend/WorkflowGeneratorModelSupport.cs

@@ -877,7 +877,7 @@ public void LoadClip3(string type, string modelA, string modelB, string modelC)
         else if (IsFlux2())
         {
             helpers.LoadClip("flux2", helpers.GetMistralFlux2Model());
-            helpers.DoVaeLoader(null, "flux-2", "flux2-vae");
+            helpers.DoVaeLoader(UserInput.SourceSession?.User?.Settings?.VAEs?.DefaultFlux2VAE, "flux-2", "flux2-vae");
         }
         else if (IsFlux() && (LoadingClip is null || LoadingVAE is null || UserInput.Get(T2IParamTypes.T5XXLModel) is not null || UserInput.Get(T2IParamTypes.ClipLModel) is not null))
         {

src/Core/Settings.cs

@@ -492,6 +492,10 @@ public class VAEsData : AutoConfiguration
             [ManualSettingsOptions(Impl = null, Vals = ["None"])]
             public string DefaultFluxVAE = "None";
 
+            [ConfigComment("What VAE to use with Flux2 models by default.")]
+            [ManualSettingsOptions(Impl = null, Vals = ["None"])]
+            public string DefaultFlux2VAE = "None";
+
             [ConfigComment("What VAE to use with SD3 models by default.")]
             [ManualSettingsOptions(Impl = null, Vals = ["None"])]
             public string DefaultSD3VAE = "None";

src/wwwroot/js/genpage/helpers/settings_editor.js

@@ -154,7 +154,7 @@ function applyThemeSetting(theme_info) {
 function loadUserSettings(callback = null) {
     genericRequest('GetUserSettings', {}, data => {
         if (coreModelMap['VAE'] != null) {
-            for (let setting of ['defaultsdxlvae', 'defaultsdv1vae', 'defaultsvdvae', 'defaultfluxvae', 'defaultsd3vae', 'defaultmochivae']) {
+            for (let setting of ['defaultsdxlvae', 'defaultsdv1vae', 'defaultsvdvae', 'defaultfluxvae', 'defaultflux2vae', 'defaultsd3vae', 'defaultmochivae']) {
                 data.settings.vaes.value[setting].values = ['None'].concat(coreModelMap['VAE']);
             }
         }