fix: adding scheduler reception and displaying

scenario-labs · Mar 7, 2024 · cc80ab3 · cc80ab3
1 parent 8ec675e
commit cc80ab3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/package/Editor/Images/ImageDataClasses.cs b/package/Editor/Images/ImageDataClasses.cs
@@ -53,6 +53,7 @@ public class Parameters
         public int width { get; set; }
         public bool hideResults { get; set; }
         public string type { get; set; }
+        public string scheduler { get; set; }
         public string prompt { get; set; }
         public string negativePrompt { get; set; }
         public int height { get; set; }

diff --git a/package/Editor/Images/Images.cs b/package/Editor/Images/Images.cs
@@ -93,7 +93,7 @@ public static void GetInferencesData(Action callback_OnDataGet = null) //why get
                             Steps = inference.parameters.numInferenceSteps,
                             Size = new Vector2(inference.parameters.width,inference.parameters.height),
                             Guidance = inference.parameters.guidance,
-                            Scheduler = "Default", //TODO : change this to reflect the scheduler used for creating this image
+                            Scheduler = inference.parameters.scheduler,
                             Seed = image.seed,
                             CreatedAt = inference.createdAt,
                             modelId = inference.modelId