From 6cc3596c56ee33c4fa4fa24c3c896e748b05e368 Mon Sep 17 00:00:00 2001
From: JT <mohnjiles@gmail.com>
Date: Sun, 10 Mar 2024 03:23:22 -0700
Subject: [PATCH] fix tests & add pony

---
 StabilityMatrix.Core/Models/Api/CivitBaseModelType.cs | 8 ++++++--
 StabilityMatrix.Core/Models/Packages/SDWebForge.cs    | 4 +++-
 2 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/StabilityMatrix.Core/Models/Api/CivitBaseModelType.cs b/StabilityMatrix.Core/Models/Api/CivitBaseModelType.cs
index 9721d717..855b075b 100644
--- a/StabilityMatrix.Core/Models/Api/CivitBaseModelType.cs
+++ b/StabilityMatrix.Core/Models/Api/CivitBaseModelType.cs
@@ -8,6 +8,9 @@ public enum CivitBaseModelType
 {
     All,
 
+    [StringValue("Pony")]
+    Pony,
+
     [StringValue("SD 1.5")]
     Sd15,
 
@@ -32,13 +35,14 @@ public enum CivitBaseModelType
     [StringValue("SDXL Lightning")]
     SdxlLightning,
 
+    [StringValue("SDXL Turbo")]
+    SdxlTurbo,
+
     [StringValue("SVD")]
     SVD,
 
     [StringValue("Stable Cascade")]
     StableCascade,
 
-    [StringValue("SDXL Turbo")]
-    SdxlTurbo,
     Other,
 }
diff --git a/StabilityMatrix.Core/Models/Packages/SDWebForge.cs b/StabilityMatrix.Core/Models/Packages/SDWebForge.cs
index f2046e5b..362e6c26 100644
--- a/StabilityMatrix.Core/Models/Packages/SDWebForge.cs
+++ b/StabilityMatrix.Core/Models/Packages/SDWebForge.cs
@@ -178,5 +178,7 @@ public class SDWebForge(
     }
 
     public override string? ExtraLaunchArguments { get; set; } =
-        $"--gradio-allowed-path \"{settingsManager.ImagesDirectory}\"";
+        settingsManager.IsLibraryDirSet
+            ? $"--gradio-allowed-path \"{settingsManager.ImagesDirectory}\""
+            : string.Empty;
 }