Merge branch 'main' of https://github.com/ionite34/StabilityMatrix

Author: mohnjiles

CHANGELOG.md

@@ -12,8 +12,10 @@ and this project adheres to [Semantic Versioning 2.0](https://semver.org/spec/v2
 - Fixed more crashes when loading invalid connected model info files
 - Fixed pip installs not parsing comments properly
 - Fixed crash when sending input to a process that isn't running
+- Fixed breadcrumb on console page showing incorrect running package name
 - Fixed [#576](https://github.com/LykosAI/StabilityMatrix/issues/576) - drag & drop crashes on macOS & Linux
 - Fixed [#594](https://github.com/LykosAI/StabilityMatrix/issues/594) - missing thumbnails in Inference model selector
+- Fixed [#600](https://github.com/LykosAI/StabilityMatrix/issues/600) - kohya_ss v24+ not launching
 - Downgraded Avalonia back to 11.0.9 to fix [#589](https://github.com/LykosAI/StabilityMatrix/issues/589) and possibly other rendering issues
 
 ## v2.10.1

StabilityMatrix.Avalonia/ViewModels/RunningPackageViewModel.cs

@@ -26,7 +26,7 @@ public partial class RunningPackageViewModel : PageViewModelBase, IDisposable, I
 
     public PackagePair RunningPackage { get; }
     public ConsoleViewModel Console { get; }
-    public override string Title => RunningPackage.InstalledPackage.PackageName ?? "Running Package";
+    public override string Title => RunningPackage.InstalledPackage.DisplayName ?? "Running Package";
     public override IconSource IconSource => new SymbolIconSource();
 
     [ObservableProperty]

StabilityMatrix.Core/Models/Packages/KohyaSs.cs

@@ -159,90 +159,7 @@ public override async Task RunPackage(
         Action<ProcessOutput>? onConsoleOutput
     )
     {
-        await SetupVenv(installedPackagePath).ConfigureAwait(false);
-
-        // update gui files to point to venv accelerate
-        await runner.RunInThreadWithLock(() =>
-        {
-            var scope = Py.CreateScope();
-            scope.Exec(
-                """
-                import ast
-
-                class StringReplacer(ast.NodeTransformer):
-                    def __init__(self, old: str, new: str, replace_count: int = -1):
-                        self.old = old
-                        self.new = new
-                        self.replace_count = replace_count
-                    
-                    def visit_Constant(self, node: ast.Constant) -> ast.Constant:
-                        if isinstance(node.value, str) and self.old in node.value:
-                            new_value = node.value.replace(self.old, self.new, self.replace_count)
-                            node.value = new_value
-                        return node
-                
-                    def rewrite_module(self, module_text: str) -> str:
-                        tree = ast.parse(module_text)
-                        tree = self.visit(tree)
-                        return ast.unparse(tree)
-                """
-            );
-
-            var replacementAcceleratePath = Compat.IsWindows
-                ? @".\venv\scripts\accelerate"
-                : "./venv/bin/accelerate";
-
-            var replacer = scope.InvokeMethod(
-                "StringReplacer",
-                "accelerate".ToPython(),
-                $"{replacementAcceleratePath}".ToPython(),
-                1.ToPython()
-            );
-
-            var kohyaGuiDir = Path.Combine(installedPackagePath, "kohya_gui");
-            var guiDirExists = Directory.Exists(kohyaGuiDir);
-            var filesToUpdate = new List<string>();
-
-            if (guiDirExists)
-            {
-                filesToUpdate.AddRange(
-                    [
-                        "lora_gui.py",
-                        "dreambooth_gui.py",
-                        "textual_inversion_gui.py",
-                        "wd14_caption_gui.py",
-                        "finetune_gui.py"
-                    ]
-                );
-            }
-            else
-            {
-                filesToUpdate.AddRange(
-                    [
-                        "lora_gui.py",
-                        "dreambooth_gui.py",
-                        "textual_inversion_gui.py",
-                        Path.Combine("library", "wd14_caption_gui.py"),
-                        "finetune_gui.py"
-                    ]
-                );
-            }
-
-            foreach (var file in filesToUpdate)
-            {
-                var path = Path.Combine(guiDirExists ? kohyaGuiDir : installedPackagePath, file);
-                if (!File.Exists(path))
-                    continue;
-
-                var text = File.ReadAllText(path);
-                if (text.Contains(replacementAcceleratePath.Replace(@"\", @"\\")))
-                    continue;
-
-                var result = replacer.InvokeMethod("rewrite_module", text.ToPython());
-                var resultStr = result.ToString();
-                File.WriteAllText(path, resultStr);
-            }
-        });
+        var venvRunner = await SetupVenvPure(installedPackagePath).ConfigureAwait(false);
 
         void HandleConsoleOutput(ProcessOutput s)
         {
@@ -262,7 +179,7 @@ void HandleConsoleOutput(ProcessOutput s)
 
         var args = $"\"{Path.Combine(installedPackagePath, command)}\" {arguments}";
 
-        VenvRunner.RunDetached(args.TrimEnd(), HandleConsoleOutput, OnExit);
+        venvRunner.RunDetached(args.TrimEnd(), HandleConsoleOutput, OnExit);
     }
 
     public override Dictionary<SharedFolderType, IReadOnlyList<string>>? SharedFolders { get; }

StabilityMatrix.Core/Models/Packages/SDWebForge.cs

@@ -139,6 +139,13 @@ public override async Task InstallPackage(
         progress?.Report(new ProgressReport(-1f, "Installing requirements...", isIndeterminate: true));
 
         var requirements = new FilePath(installLocation, "requirements_versions.txt");
+        var requirementsContent = await requirements.ReadAllTextAsync().ConfigureAwait(false);
+        if (!requirementsContent.Contains("pydantic"))
+        {
+            requirementsContent += "pydantic==1.10.15";
+            await requirements.WriteAllTextAsync(requirementsContent).ConfigureAwait(false);
+        }
+
         var pipArgs = new PipInstallArgs();
         if (torchVersion is TorchVersion.DirectMl)
         {
@@ -161,10 +168,7 @@ public override async Task InstallPackage(
                 );
         }
 
-        pipArgs = pipArgs.WithParsedFromRequirementsTxt(
-            await requirements.ReadAllTextAsync().ConfigureAwait(false),
-            excludePattern: "torch"
-        );
+        pipArgs = pipArgs.WithParsedFromRequirementsTxt(requirementsContent, excludePattern: "torch");
 
         await venvRunner.PipInstall(pipArgs, onConsoleOutput).ConfigureAwait(false);
         progress?.Report(new ProgressReport(1f, "Install complete", isIndeterminate: false));

StabilityMatrix.Core/Python/PyVenvRunner.cs

@@ -545,7 +545,8 @@ public void RunDetached(
         if (Compat.IsWindows)
         {
             var portableGitBin = GlobalConfig.LibraryDir.JoinDir("PortableGit", "bin");
-            env["PATH"] = Compat.GetEnvPathWithExtensions(portableGitBin);
+            var venvBin = RootPath.JoinDir(RelativeBinPath);
+            env["PATH"] = Compat.GetEnvPathWithExtensions(portableGitBin, venvBin);
             env["GIT"] = portableGitBin.JoinFile("git.exe");
         }