From 8553cd18f045589758e5f810c6f67a9fbd058009 Mon Sep 17 00:00:00 2001
From: Ionite <dev@ionite.io>
Date: Fri, 13 Dec 2024 22:18:02 -0500
Subject: [PATCH] Fix JsonValue deserialize error with Inference

---
 StabilityMatrix.Core/Models/Api/Comfy/ComfyInputInfo.cs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/StabilityMatrix.Core/Models/Api/Comfy/ComfyInputInfo.cs b/StabilityMatrix.Core/Models/Api/Comfy/ComfyInputInfo.cs
index c69f14d2..fd56fb6f 100644
--- a/StabilityMatrix.Core/Models/Api/Comfy/ComfyInputInfo.cs
+++ b/StabilityMatrix.Core/Models/Api/Comfy/ComfyInputInfo.cs
@@ -7,10 +7,10 @@ namespace StabilityMatrix.Core.Models.Api.Comfy;
 public class ComfyInputInfo
 {
     [JsonPropertyName("required")]
-    public Dictionary<string, JsonValue>? Required { get; set; }
+    public Dictionary<string, JsonNode>? Required { get; set; }
 
     [JsonPropertyName("optional")]
-    public Dictionary<string, JsonValue>? Optional { get; set; }
+    public Dictionary<string, JsonNode>? Optional { get; set; }
 
     public List<string>? GetRequiredValueAsNestedList(string key)
     {