Fix batch count seed increments not being recorded in project or image metadata

Author: mohnjiles

CHANGELOG.md

@@ -65,6 +65,7 @@ and this project adheres to [Semantic Versioning 2.0](https://semver.org/spec/v2
 ### Fixed
 - Fixed incorrect IPAdapter download links in the HuggingFace model browser
 - Fixed potential memory leak of transient controls (Inference Prompt and Output Image Viewer) not being garbage collected due to event subscriptions
+- Fixed Batch Count seeds not being recorded properly in Inference projects and image metadata
 
 ## v2.11.5
 ### Added

StabilityMatrix.Avalonia/ViewModels/Inference/InferenceTextToImageViewModel.cs

@@ -242,13 +242,20 @@ CancellationToken cancellationToken
             var buildPromptArgs = new BuildPromptEventArgs { Overrides = overrides, SeedOverride = seed };
             BuildPrompt(buildPromptArgs);
 
+            // update seed in project for batches
+            var inferenceProject = InferenceProjectDocument.FromLoadable(this);
+            if (inferenceProject.State?["Seed"]?["Seed"] is not null)
+            {
+                inferenceProject = inferenceProject.WithState(x => x["Seed"]["Seed"] = seed);
+            }
+
             var generationArgs = new ImageGenerationEventArgs
             {
                 Client = ClientManager.Client,
                 Nodes = buildPromptArgs.Builder.ToNodeDictionary(),
                 OutputNodeNames = buildPromptArgs.Builder.Connections.OutputNodeNames.ToArray(),
-                Parameters = SaveStateToParameters(new GenerationParameters()),
-                Project = InferenceProjectDocument.FromLoadable(this),
+                Parameters = SaveStateToParameters(new GenerationParameters(), Convert.ToUInt64(seed)),
+                Project = inferenceProject,
                 FilesToTransfer = buildPromptArgs.FilesToTransfer,
                 BatchIndex = i,
                 // Only clear output images on the first batch
@@ -281,13 +288,16 @@ public void LoadStateFromParameters(GenerationParameters parameters)
     }
 
     /// <inheritdoc />
-    public GenerationParameters SaveStateToParameters(GenerationParameters parameters)
+    public GenerationParameters SaveStateToParameters(GenerationParameters parameters) =>
+        SaveStateToParameters(parameters, null);
+
+    private GenerationParameters SaveStateToParameters(GenerationParameters parameters, ulong? seedOverride)
     {
         parameters = PromptCardViewModel.SaveStateToParameters(parameters);
         parameters = SamplerCardViewModel.SaveStateToParameters(parameters);
         parameters = ModelCardViewModel.SaveStateToParameters(parameters);
 
-        parameters.Seed = (ulong)SeedCardViewModel.Seed;
+        parameters.Seed = seedOverride ?? (ulong)SeedCardViewModel.Seed;
 
         return parameters;
     }