From 8fdd737c51e76754501b63656d5f56a2c72b1269 Mon Sep 17 00:00:00 2001
From: JT <mohnjiles@gmail.com>
Date: Tue, 12 Mar 2024 19:07:05 -0700
Subject: [PATCH] Fix paths for new kohya directory structure

---
 StabilityMatrix.Core/Models/Packages/KohyaSs.cs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/StabilityMatrix.Core/Models/Packages/KohyaSs.cs b/StabilityMatrix.Core/Models/Packages/KohyaSs.cs
index 46687b35..5db9c583 100644
--- a/StabilityMatrix.Core/Models/Packages/KohyaSs.cs
+++ b/StabilityMatrix.Core/Models/Packages/KohyaSs.cs
@@ -206,13 +206,13 @@ public class KohyaSs(
                 "lora_gui.py",
                 "dreambooth_gui.py",
                 "textual_inversion_gui.py",
-                Path.Combine("library", "wd14_caption_gui.py"),
+                "wd14_caption_gui.py",
                 "finetune_gui.py"
             };
 
             foreach (var file in filesToUpdate)
             {
-                var path = Path.Combine(installedPackagePath, file);
+                var path = Path.Combine(installedPackagePath, "kohya_gui", file);
                 var text = File.ReadAllText(path);
                 if (text.Contains(replacementAcceleratePath.Replace(@"\", @"\\")))
                     continue;