Add TiledVAEDecode backend integration (ComfyNodeBuilder + ComfyClient)

Author: Domica

StabilityMatrix.Core/Inference/ComfyClient.cs

@@ -326,22 +326,49 @@ public override async Task CloseAsync(CancellationToken cancellationToken = defa
         await webSocketClient.Stop(WebSocketCloseStatus.NormalClosure, string.Empty).ConfigureAwait(false);
     }
 
-    public async Task<ComfyTask> QueuePromptAsync(
-        Dictionary<string, ComfyNode> nodes,
-        CancellationToken cancellationToken = default
-    )
+public async Task<ComfyTask> QueuePromptAsync(
+    Dictionary<string, ComfyNode> nodes,
+    CancellationToken cancellationToken = default
+)
+{
+    var request = new ComfyPromptRequest { ClientId = ClientId, Prompt = nodes };
+
+    // DEBUG: dump final workflow JSON
+    try
     {
-        var request = new ComfyPromptRequest { ClientId = ClientId, Prompt = nodes };
-        var result = await comfyApi.PostPrompt(request, cancellationToken).ConfigureAwait(false);
+        var json = JsonSerializer.Serialize(request, jsonSerializerOptions);
 
-        // Add task to dictionary and set it as the current task
-        var task = new ComfyTask(result.PromptId);
-        PromptTasks.TryAdd(result.PromptId, task);
-        currentPromptTask = task;
+        var debugDir = Path.Combine(
+            Environment.GetFolderPath(Environment.SpecialFolder.ApplicationData),
+            "StabilityMatrix",
+            "Debug"
+        );
+
+        Directory.CreateDirectory(debugDir);
+
+        var path = Path.Combine(
+            debugDir,
+            $"wan_workflow_debug_request.json"
+        );
 
-        return task;
+        File.WriteAllText(path, json);
+
+        Logger.Warn("WAN DEBUG: Dumped final request JSON to {0}", path);
+    }
+    catch (Exception ex)
+    {
+        Logger.Error(ex, "WAN DEBUG: Failed to dump final request JSON");
     }
 
+    var result = await comfyApi.PostPrompt(request, cancellationToken).ConfigureAwait(false);
+
+    var task = new ComfyTask(result.PromptId);
+    PromptTasks.TryAdd(result.PromptId, task);
+    currentPromptTask = task;
+
+    return task;
+}
+
     public async Task InterruptPromptAsync(CancellationToken cancellationToken = default)
     {
         await comfyApi.PostInterrupt(cancellationToken).ConfigureAwait(false);

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

@@ -1,4 +1,4 @@
-using System.ComponentModel;
+using System.ComponentModel;
 using System.ComponentModel.DataAnnotations;
 using System.Diagnostics.CodeAnalysis;
 using System.Drawing;
@@ -1097,7 +1097,34 @@ public record NRS : ComfyTypedNodeBase<ModelNodeConnection>
         public required double Squash { get; set; }
     }
 
-    public ImageNodeConnection Lambda_LatentToImage(LatentNodeConnection latent, VAENodeConnection vae)
+   public ImageNodeConnection Lambda_LatentToImage(
+    LatentNodeConnection latent, 
+    VAENodeConnection vae,
+    bool useTiledVAE = false,
+    int tileSize = 512,
+    int overlap = 64,
+    int temporalSize = 64,
+    int temporalOverlap = 8)
+{
+    if (useTiledVAE)
+    {
+        var name = GetUniqueName("VAEDecodeTiled");
+        return Nodes
+            .AddTypedNode(
+                new TiledVAEDecode
+                {
+                    Name = name,
+                    Samples = latent,
+                    Vae = vae,
+                    TileSize = tileSize,
+                    Overlap = overlap,
+                    TemporalSize = temporalSize,
+                    TemporalOverlap = temporalOverlap
+                }
+            )
+            .Output;
+    }
+    else
     {
         var name = GetUniqueName("VAEDecode");
         return Nodes
@@ -1111,6 +1138,7 @@ public ImageNodeConnection Lambda_LatentToImage(LatentNodeConnection latent, VAE
             )
             .Output;
     }
+}
 
     public LatentNodeConnection Lambda_ImageToLatent(ImageNodeConnection pixels, VAENodeConnection vae)
     {
@@ -1519,11 +1547,28 @@ public ImageNodeConnection GetPrimaryAsImage()
     /// <summary>
     /// Get or convert latest primary connection to image
     /// </summary>
+<<<<<<< HEAD
     public ImageNodeConnection GetPrimaryAsImage(PrimaryNodeConnection primary, VAENodeConnection vae)
     {
         return primary.Match(latent => Lambda_LatentToImage(latent, vae), image => image);
     }
 
+=======
+    public ImageNodeConnection GetPrimaryAsImage(
+    PrimaryNodeConnection primary, 
+    VAENodeConnection vae,
+    bool useTiledVAE = false,
+    int tileSize = 512,
+    int overlap = 64,
+    int temporalSize = 64,
+    int temporalOverlap = 8)
+{
+    return primary.Match(
+        latent => Lambda_LatentToImage(latent, vae, useTiledVAE, tileSize, overlap, temporalSize, temporalOverlap), 
+        image => image
+    );
+    }
+>>>>>>> 27accf16 (Update ComfyNodeBuilder.cs)
     /// <summary>
     /// Get or convert latest primary connection to image
     /// </summary>
@@ -1640,4 +1685,4 @@ public VAENodeConnection GetDefaultVAE()
     }
 
     public NodeBuilderConnections Connections { get; } = new();
-}
+}