Add TiledVAEDecode backend integration (ComfyNodeBuilder + ComfyClient)

Author: Domica

StabilityMatrix.Avalonia/Controls/Inference/TiledVAECard.axaml

@@ -5,15 +5,16 @@
     xmlns:ui="clr-namespace:FluentAvalonia.UI.Controls;assembly=FluentAvalonia"
     xmlns:vmInference="clr-namespace:StabilityMatrix.Avalonia.ViewModels.Inference"
     x:DataType="vmInference:TiledVAECardViewModel">
+
     <Design.PreviewWith>
         <controls:TiledVAECard />
     </Design.PreviewWith>
 
     <Style Selector="controls|TiledVAECard">
-        <!--  Set Defaults  -->
         <Setter Property="Template">
             <ControlTemplate>
                 <controls:Card x:Name="PART_Card">
+
                     <controls:Card.Styles>
                         <Style Selector="ui|NumberBox">
                             <Setter Property="Margin" Value="12,0,0,0" />
@@ -25,8 +26,10 @@
                             <Setter Property="SpinButtonPlacementMode" Value="Inline" />
                         </Style>
                     </controls:Card.Styles>
+
                     <StackPanel Spacing="8">
-                        <!--  Tile Size  -->
+
+                        <!-- Tile Size -->
                         <Grid ColumnDefinitions="Auto,*">
                             <TextBlock
                                 Grid.Column="0"
@@ -37,7 +40,7 @@
                                 Value="{Binding TileSize, Mode=TwoWay}" />
                         </Grid>
 
-                        <!--  Overlap  -->
+                        <!-- Overlap -->
                         <Grid ColumnDefinitions="Auto,*">
                             <TextBlock
                                 Grid.Column="0"
@@ -48,29 +51,40 @@
                                 Value="{Binding Overlap, Mode=TwoWay}" />
                         </Grid>
 
-                        <!--  Temporal Size (for Video VAEs)  -->
-                        <Grid ColumnDefinitions="Auto,*">
-                            <TextBlock
-                                Grid.Column="0"
-                                VerticalAlignment="Center"
-                                Text="Temporal Size" />
-                            <ui:NumberBox
-                                Grid.Column="1"
-                                Value="{Binding TemporalSize, Mode=TwoWay}" />
-                        </Grid>
+                        <!-- Enable Temporal Tiling -->
+                        <ToggleSwitch
+                            Header="Enable Temporal Tiling"
+                            IsChecked="{Binding UseTemporalTiling}" />
+
+                        <!-- Temporal Controls (Visible only when enabled) -->
+                        <StackPanel IsVisible="{Binding UseTemporalTiling}" Spacing="8">
+
+                            <!-- Temporal Size -->
+                            <Grid ColumnDefinitions="Auto,*">
+                                <TextBlock
+                                    Grid.Column="0"
+                                    VerticalAlignment="Center"
+                                    Text="Temporal Size" />
+                                <ui:NumberBox
+                                    Grid.Column="1"
+                                    Value="{Binding TemporalSize, Mode=TwoWay}" />
+                            </Grid>
+
+                            <!-- Temporal Overlap -->
+                            <Grid ColumnDefinitions="Auto,*">
+                                <TextBlock
+                                    Grid.Column="0"
+                                    VerticalAlignment="Center"
+                                    Text="Temporal Overlap" />
+                                <ui:NumberBox
+                                    Grid.Column="1"
+                                    Value="{Binding TemporalOverlap, Mode=TwoWay}"
+                                    SmallChange="4"
+                                    LargeChange="16" />
+                            </Grid>
+
+                        </StackPanel>
 
-                        <!--  Temporal Overlap (for Video VAEs)  -->
-                        <Grid ColumnDefinitions="Auto,*">
-                            <TextBlock
-                                Grid.Column="0"
-                                VerticalAlignment="Center"
-                                Text="Temporal Overlap" />
-                            <ui:NumberBox
-                                Grid.Column="1"
-                                Value="{Binding TemporalOverlap, Mode=TwoWay}"
-                                SmallChange="4"
-                                LargeChange="16" />
-                        </Grid>
                     </StackPanel>
                 </controls:Card>
             </ControlTemplate>

StabilityMatrix.Avalonia/ViewModels/Inference/InferenceWanTextToVideoViewModel.cs

@@ -10,6 +10,7 @@
 using StabilityMatrix.Core.Attributes;
 using StabilityMatrix.Core.Models;
 using StabilityMatrix.Core.Services;
+git status
 
 namespace StabilityMatrix.Avalonia.ViewModels.Inference;
 
@@ -115,8 +116,14 @@ protected override void BuildPrompt(BuildPromptEventArgs args)
 
         SamplerCardViewModel.ApplyStep(args);
 
-        // Animated webp output
+        // ADD THIS - Apply TiledVAE module
+        TiledVAEModule.ApplyStep(args);
+
+        // ADD THIS LINE - Invoke pre-output actions BEFORE video output!
+        args.Builder.InvokeAllPreOutputActions();
+
         VideoOutputSettingsCardViewModel.ApplyStep(args);
+
     }
 
     /// <inheritdoc />

StabilityMatrix.Avalonia/ViewModels/Inference/Modules/TiledVAEModule.cs

@@ -34,8 +34,14 @@ protected override void OnApplyStep(ModuleApplyStepEventArgs e)
             var latent = builder.Connections.Primary.AsT0;
             var vae = builder.Connections.GetDefaultVAE();
 
-            // Use tiled VAE decode instead of standard decode
-            var tiledDecode = builder.Nodes.AddTypedNode(
+            logger.LogWarning("TiledVAE: Injecting TiledVAEDecode");
+            logger.LogWarning("UseCustomTemporalTiling value at runtime: {Value}", card.UseCustomTemporalTiling);
+
+            // Always valid values (Wan requires temporal tiling)
+            int temporalSize = card.UseCustomTemporalTiling ? card.TemporalSize : 64;
+            int temporalOverlap = card.UseCustomTemporalTiling ? card.TemporalOverlap : 8;
+
+            var node = builder.Nodes.AddTypedNode(
                 new ComfyNodeBuilder.TiledVAEDecode
                 {
                     Name = builder.Nodes.GetUniqueName("TiledVAEDecode"),

StabilityMatrix.Avalonia/ViewModels/Inference/TiledVAECardViewModel.cs

@@ -14,24 +14,35 @@ public partial class TiledVAECardViewModel : LoadableViewModelBase
 {
     public const string ModuleKey = "TiledVAE";
 
+    // Spatial tile size (valid for Wan)
     [ObservableProperty]
     [NotifyDataErrorInfo]
     [Required]
     [Range(64, 4096)]
     private int tileSize = 512;
 
+    // Spatial overlap
     [ObservableProperty]
     [NotifyDataErrorInfo]
     [Required]
     [Range(0, 4096)]
     private int overlap = 64;
 
+    // NEW: Toggle now means "Use custom temporal tiling settings"
+    [ObservableProperty]
+    private bool useCustomTemporalTiling = false;
+
+    // Temporal tile size (must be >= 8)
+    [ObservableProperty]
+    private bool useTemporalTiling = true;
+
     [ObservableProperty]
     [NotifyDataErrorInfo]
     [Required]
     [Range(8, 4096)]
     private int temporalSize = 64;
 
+    // Temporal overlap (must be >= 4)
     [ObservableProperty]
     [NotifyDataErrorInfo]
     [Required]

StabilityMatrix.Avalonia/ViewModels/Settings/InferenceSettingsViewModel.cs

@@ -69,6 +69,8 @@ public partial class InferenceSettingsViewModel : PageViewModelBase
     [ObservableProperty]
     private bool filterExtraNetworksByBaseModel;
 
+    [ObservableProperty] private bool enableTiledVae;
+
     private List<string> ignoredFileNameFormatVars =
     [
         "author",
@@ -191,6 +193,12 @@ ISettingsManager settingsManager
             settings => settings.InferenceDimensionStepChange,
             true
         );
+        settingsManager.RelayPropertyFor(
+            this,
+            vm => vm.EnableTiledVae,
+            settings => settings.EnableTiledVae,
+            true
+        );
 
         FavoriteDimensions
             .ToObservableChangeSet()

StabilityMatrix.Core/Models/Api/Comfy/Nodes/TiledVAEDecodeVideoNode.cs

@@ -0,0 +1,20 @@
+namespace StabilityMatrix.Core.Models.Api.Comfy.Nodes;
+
+public record TiledVAEDecodeVideoNode : ComfyNode
+{
+    public override string ClassType => "TiledVAEDecodeVideo";
+
+    public record Inputs(
+        ComfyNodeConnection Vae,
+        ComfyNodeConnection VideoLatent,
+        int TileSize,
+        int Overlap
+    );
+
+    public Inputs Input { get; init; } = new(
+        Vae: new(),
+        VideoLatent: new(),
+        TileSize: 256,
+        Overlap: 32
+    );
+}

StabilityMatrix.Core/Models/Settings/Settings.cs

@@ -100,6 +100,9 @@ public InstalledPackage? PreferredWorkflowPackage
     /// </summary>
     public bool IsCompletionRemoveUnderscoresEnabled { get; set; } = true;
 
+    /// COPILOT
+    public bool EnableTiledVae { get; set; } = false;
+    
     /// <summary>
     /// Format for Inference output image file names
     /// </summary>

StabilityMatrix.Core/Settings/Inference/WanSettings.cs

@@ -0,0 +1,6 @@
+namespace StabilityMatrix.Core.Settings.Inference;
+
+public class WanSettings
+{
+    public bool EnableTiledVae { get; set; } = false;
+}