Limit video output conversions to just 3 formats that we know will wo…

…rk with comfyUI agent.
ServiceStack · Oct 9, 2024 · bd58eb5 · bd58eb5
1 parent 5378462
commit bd58eb5
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 73 deletions.
diff --git a/AiServer.ServiceModel/QueueMediaTransforms.cs b/AiServer.ServiceModel/QueueMediaTransforms.cs
@@ -375,14 +375,8 @@ public enum ConvertVideoOutputFormat
     MP4,
     [EnumMember(Value = "avi")]
     AVI,
-    [EnumMember(Value = "mkv")]
-    MKV,
     [EnumMember(Value = "mov")]
-    MOV,
-    [EnumMember(Value = "webm")]
-    WebM,
-    [EnumMember(Value = "gif")]
-    GIF
+    MOV
 }
 
 [DataContract]

diff --git a/AiServer.Tests/ComfyAdminTasks.cs b/AiServer.Tests/ComfyAdminTasks.cs
@@ -12,70 +12,6 @@ public class ComfyAdminTasks
 {
     private static bool useLocal = true;
     private ConfigureSecrets ConfigureSecrets = new();
-
-    public static Dictionary<string, ComfyApiModelSettings> ImportCivitAiModelSettings = new()
-    {
-        {
-            "https://civitai.com/models/194768/jib-mix-realistic-xl?modelVersionId=610292",
-            new ComfyApiModelSettings
-            {
-                Height = 1024,
-                Width = 1024,
-                Sampler = ComfySampler.euler_ancestral,
-                Scheduler = "karras",
-                CfgScale = 6.0,
-                Steps = 18
-            }
-        },
-        {
-            "https://civitai.com/models/553410/crazycaricaturesxl?modelVersionId=615871",
-            new ComfyApiModelSettings
-            {
-                Height = 1024,
-                Width = 1024,
-                Sampler = ComfySampler.euler_ancestral,
-                Scheduler = "normal",
-                CfgScale = 8.0,
-                Steps = 18
-            }
-        },
-        {
-            "https://civitai.com/models/129403?modelVersionId=259194",
-            new ComfyApiModelSettings
-            {
-                Height = 1024,
-                Width = 1024,
-                Sampler = ComfySampler.euler_ancestral,
-                Scheduler = "normal",
-                CfgScale = 3.0,
-                Steps = 12
-            }
-        },
-        {
-            "https://civitai.com/models/157665/lah-hongchen-or-sdxl-and-sd15",
-            new ComfyApiModelSettings
-            {
-                Height = 1024,
-                Width = 1024,
-                Sampler = ComfySampler.euler,
-                Scheduler = "normal",
-                CfgScale = 1.0,
-                Steps = 8
-            }
-        },
-        {
-            "https://civitai.com/models/350352?modelVersionId=391971",
-            new ComfyApiModelSettings
-            {
-                Height = 1024,
-                Width = 1024,
-                Sampler = ComfySampler.euler,
-                Scheduler = "normal",
-                CfgScale = 1.0,
-                Steps = 8
-            }
-        }
-    };
 
     private static List<CreateMediaProvider> MediaProviders = new List<CreateMediaProvider>
     {
@@ -88,7 +24,7 @@ public class ComfyAdminTasks
             HeartbeatUrl = "https://api.replicate.com/",
             ApiBaseUrl = "https://api.replicate.com/",
             Models = new List<string> { "flux1-dev","flux1-schnell" },
-            MediaTypeId = 1
+            MediaTypeId = "ConfyUI"
         },
         new()
         {
@@ -105,7 +41,7 @@ public class ComfyAdminTasks
                 "animexlXuebimix_v60LCM.safetensors",
                 "LahHongchenSDXLSD15_xlLightning.safetensors"
             },
-            MediaTypeId = 2
+            MediaTypeId = "ComfyUI"
         }
     };