only send preview metadata once

2024-09-22 23:44:41 +09:00
parent 5bf4c2f6bd
commit ccbb7c3320
1 changed files with 8 additions and 3 deletions
--- a/src/BuiltinExtensions/ComfyUIBackend/ComfyUIAPIAbstractBackend.cs
+++ b/src/BuiltinExtensions/ComfyUIBackend/ComfyUIAPIAbstractBackend.cs
@@ -227,13 +227,18 @@ public abstract class ComfyUIAPIAbstractBackend : AbstractT2IBackend
        float curPercent = 0;
        void yieldProgressUpdate()
        {
-            takeOutput(new JObject()
+            JObject toSend = new()
            {
                ["batch_index"] = batchId,
-                ["metadata"] = previewMetadata,
                ["overall_percent"] = nodesDone / (float)expectedNodes,
                ["current_percent"] = curPercent
-            });
+            };
+            if (previewMetadata is not null)
+            {
+                toSend["metadata"] = previewMetadata;
+                previewMetadata = null;
+            }
+            takeOutput(toSend);
        }
        try
        {