Merge pull request #629 from LykosAI/main

v2.10.3

CHANGELOG.md

@@ -5,6 +5,18 @@ All notable changes to Stability Matrix will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.1.0/),
 and this project adheres to [Semantic Versioning 2.0](https://semver.org/spec/v2.0.0.html).
 
+## v2.10.3
+### Changed
+- Centered OpenArt browser cards
+### Fixed
+- Fixed MPS install on macOS for ComfyUI, A1111, SDWebUI Forge, and SDWebUI UX causing torch to be upgraded to dev nightly versions and causing incompatibilities with dependencies.
+- Fixed crash when failing to parse Python package details
+- Fixed "Auto Scroll to End" not working in some scenarios
+- Fixed "Auto Scroll to End" toggle button not scrolling to the end when toggled on
+- Fixed/reverted output folder name changes for Automatic1111
+- Fixed xformers being uninstalled with every ComfyUI update
+- Fixed missing progress text during package installs
+
 ## v2.10.2
 ### Changed
 - Updated translations for Spanish and Turkish

StabilityMatrix.Avalonia/ViewModels/Dialogs/PythonPackagesItemViewModel.cs

@@ -6,6 +6,7 @@ using Avalonia.Controls;
 using CommunityToolkit.Mvvm.ComponentModel;
 using Semver;
 using StabilityMatrix.Avalonia.ViewModels.Base;
+using StabilityMatrix.Core.Exceptions;
 using StabilityMatrix.Core.Helper;
 using StabilityMatrix.Core.Models.FileInterfaces;
 using StabilityMatrix.Core.Python;
@@ -114,6 +115,10 @@ public partial class PythonPackagesItemViewModel : ViewModelBase
                 }
             }
         }
+        catch (ProcessException)
+        {
+            // Ignore
+        }
         finally
         {
             IsLoading = false;

StabilityMatrix.Avalonia/ViewModels/Progress/PackageInstallProgressItemViewModel.cs

@@ -43,7 +43,7 @@ public class PackageInstallProgressItemViewModel : ProgressItemViewModelBase
     private void PackageModificationRunnerOnProgressChanged(object? sender, ProgressReport e)
     {
         Progress.Value = e.Percentage;
-        Progress.Description = e.Message;
+        Progress.Description = e.ProcessOutput?.Text ?? e.Message;
         Progress.IsIndeterminate = e.IsIndeterminate;
         Progress.Text = packageModificationRunner.CurrentStep?.ProgressTitle;
         Name = packageModificationRunner.CurrentStep?.ProgressTitle;

StabilityMatrix.Avalonia/ViewModels/RunningPackageViewModel.cs

@@ -74,6 +74,14 @@ public partial class RunningPackageViewModel : PageViewModelBase, IDisposable, I
         );
     }
 
+    public override void OnLoaded()
+    {
+        if (AutoScrollToEnd)
+        {
+            EventManager.Instance.OnScrollToBottomRequested();
+        }
+    }
+
     private void BasePackageOnExited(object? sender, int e)
     {
         IsRunning = false;
@@ -104,6 +112,14 @@ public partial class RunningPackageViewModel : PageViewModelBase, IDisposable, I
         }
     }
 
+    partial void OnAutoScrollToEndChanged(bool value)
+    {
+        if (value)
+        {
+            EventManager.Instance.OnScrollToBottomRequested();
+        }
+    }
+
     [RelayCommand]
     private async Task Restart()
     {

StabilityMatrix.Avalonia/Views/CivitAiBrowserPage.axaml

@@ -33,7 +33,7 @@
                 </Transitions>
             </Setter>
             
-            <Style Selector="^ labs|AsyncImage">
+            <Style Selector="^ vendorLabs|BetterAsyncImage">
                 <Setter Property="Transitions">
                     <Transitions>
                         <TransformOperationsTransition Property="RenderTransform"
@@ -49,7 +49,7 @@
             <Style Selector="^:pointerover">
                 <Setter Property="BoxShadow" Value="0 0 40 0 #60000000"/>
                 <Setter Property="Cursor" Value="Hand" />
-                <Style Selector="^ labs|AsyncImage">
+                <Style Selector="^ vendorLabs|BetterAsyncImage">
                     <Setter Property="CornerRadius" Value="12"/>
                     <Setter Property="RenderTransform" Value="scale(1.03, 1.03)"/>
                 </Style>
@@ -61,7 +61,7 @@
             <Style Selector="^:not(:pointerover)">
                 <Setter Property="BoxShadow" Value="0 0 20 0 #60000000"/>
                 <Setter Property="Cursor" Value="Arrow" />
-                <Style Selector="^ labs|AsyncImage">
+                <Style Selector="^ vendorLabs|BetterAsyncImage">
                     <Setter Property="CornerRadius" Value="8"/>
                     <Setter Property="RenderTransform" Value="scale(1, 1)"/>
                 </Style>

StabilityMatrix.Avalonia/Views/ConsoleOutputPage.axaml.cs

@@ -36,8 +36,8 @@ public partial class ConsoleOutputPage : UserControlBase
 
     protected override void OnLoaded(RoutedEventArgs e)
     {
-        base.OnLoaded(e);
         EventManager.Instance.ScrollToBottomRequested += OnScrollToBottomRequested;
+        base.OnLoaded(e);
     }
 
     private void OnScrollToBottomRequested(object? sender, EventArgs e)

StabilityMatrix.Avalonia/Views/OpenArtBrowserPage.axaml

@@ -305,6 +305,7 @@
 
         <ScrollViewer Grid.Row="2"
                       ScrollChanged="ScrollViewer_OnScrollChanged"
+                      HorizontalContentAlignment="Center"
                       IsVisible="{Binding !IsLoading}">
             <ItemsRepeater ItemsSource="{Binding SearchResults}"
                            ItemTemplate="{StaticResource OpenArtWorkflowTemplate}">

StabilityMatrix.Core/Models/Packages/A3WebUI.cs

@@ -73,13 +73,13 @@ public class A3WebUI(
     public override Dictionary<SharedOutputType, IReadOnlyList<string>>? SharedOutputFolders =>
         new()
         {
-            [SharedOutputType.Extras] = new[] { "output/extras-images" },
+            [SharedOutputType.Extras] = new[] { "outputs/extras-images" },
             [SharedOutputType.Saved] = new[] { "log/images" },
-            [SharedOutputType.Img2Img] = new[] { "output/img2img-images" },
-            [SharedOutputType.Text2Img] = new[] { "output/txt2img-images" },
-            [SharedOutputType.Img2ImgGrids] = new[] { "output/img2img-grids" },
-            [SharedOutputType.Text2ImgGrids] = new[] { "output/txt2img-grids" },
-            [SharedOutputType.SVD] = new[] { "output/svd" }
+            [SharedOutputType.Img2Img] = new[] { "outputs/img2img-images" },
+            [SharedOutputType.Text2Img] = new[] { "outputs/txt2img-images" },
+            [SharedOutputType.Img2ImgGrids] = new[] { "outputs/img2img-grids" },
+            [SharedOutputType.Text2ImgGrids] = new[] { "outputs/txt2img-grids" },
+            [SharedOutputType.SVD] = new[] { "outputs/svd" }
         };
 
     [SuppressMessage("ReSharper", "ArrangeObjectCreationWhenTypeNotEvident")]
@@ -186,7 +186,7 @@ public class A3WebUI(
 
     public override string MainBranch => "master";
 
-    public override string OutputFolderName => "output";
+    public override string OutputFolderName => "outputs";
 
     public override IPackageExtensionManager ExtensionManager => new A3WebUiExtensionManager(this);
 
@@ -223,7 +223,7 @@ public class A3WebUI(
                     TorchVersion.Cpu => "cpu",
                     TorchVersion.Cuda => "cu121",
                     TorchVersion.Rocm => "rocm5.6",
-                    TorchVersion.Mps => "nightly/cpu",
+                    TorchVersion.Mps => "cpu",
                     _ => throw new ArgumentOutOfRangeException(nameof(torchVersion), torchVersion, null)
                 }
             )

StabilityMatrix.Core/Models/Packages/ComfyUI.cs

@@ -209,8 +209,7 @@ public class ComfyUI(
         pipArgs = torchVersion switch
         {
             TorchVersion.DirectMl => pipArgs.WithTorchDirectML(),
-            TorchVersion.Mps
-                => pipArgs.AddArg("--pre").WithTorch().WithTorchVision().WithTorchExtraIndex("nightly/cpu"),
+            TorchVersion.Mps => pipArgs.WithTorch().WithTorchVision().WithTorchExtraIndex("cpu"),
             _
                 => pipArgs
                     .AddArg("--upgrade")
@@ -246,7 +245,6 @@ public class ComfyUI(
             excludePattern: "torch"
         );
 
-        await venvRunner.PipUninstall("xformers", onConsoleOutput).ConfigureAwait(false);
         await venvRunner.PipInstall(pipArgs, onConsoleOutput).ConfigureAwait(false);
 
         progress?.Report(new ProgressReport(1, "Installed Package Requirements", isIndeterminate: false));

StabilityMatrix.Core/Models/Packages/SDWebForge.cs

@@ -40,6 +40,17 @@ public class SDWebForge(
     public override IPackageExtensionManager ExtensionManager => null;
     public override string OutputFolderName => "output";
     public override PackageDifficulty InstallerSortOrder => PackageDifficulty.ReallyRecommended;
+    public override Dictionary<SharedOutputType, IReadOnlyList<string>>? SharedOutputFolders =>
+        new()
+        {
+            [SharedOutputType.Extras] = new[] { "output/extras-images" },
+            [SharedOutputType.Saved] = new[] { "log/images" },
+            [SharedOutputType.Img2Img] = new[] { "output/img2img-images" },
+            [SharedOutputType.Text2Img] = new[] { "output/txt2img-images" },
+            [SharedOutputType.Img2ImgGrids] = new[] { "output/img2img-grids" },
+            [SharedOutputType.Text2ImgGrids] = new[] { "output/txt2img-grids" },
+            [SharedOutputType.SVD] = new[] { "output/svd" }
+        };
 
     public override List<LaunchOptionDefinition> LaunchOptions =>
         [
@@ -162,7 +173,7 @@ public class SDWebForge(
                         TorchVersion.Cpu => "cpu",
                         TorchVersion.Cuda => "cu121",
                         TorchVersion.Rocm => "rocm5.6",
-                        TorchVersion.Mps => "nightly/cpu",
+                        TorchVersion.Mps => "cpu",
                         _ => throw new ArgumentOutOfRangeException(nameof(torchVersion), torchVersion, null)
                     }
                 );

StabilityMatrix.Core/Models/Packages/StableDiffusionUx.cs

@@ -211,7 +211,7 @@ public class StableDiffusionUx(
                         new PipInstallArgs()
                             .WithTorch("==2.1.2")
                             .WithTorchVision()
-                            .WithTorchExtraIndex("nightly/cpu"),
+                            .WithTorchExtraIndex("cpu"),
                         onConsoleOutput
                     )
                     .ConfigureAwait(false);

StabilityMatrix.Core/Python/PipShowResult.cs

@@ -1,4 +1,5 @@
-using StabilityMatrix.Core.Extensions;
+using System.Diagnostics;
+using StabilityMatrix.Core.Extensions;
 
 namespace StabilityMatrix.Core.Python;
 
@@ -31,6 +32,20 @@ public record PipShowResult
             .SplitLines(StringSplitOptions.TrimEntries | StringSplitOptions.RemoveEmptyEntries)
             // Filter warning lines
             .Where(line => !line.StartsWith("WARNING", StringComparison.OrdinalIgnoreCase))
+            .ToList();
+
+        var indexOfLicense = GetIndexBySubstring(lines, "License:");
+        var indexOfLocation = GetIndexBySubstring(lines, "Location:");
+
+        var licenseText =
+            indexOfLicense == -1 ? null : string.Join('\n', lines[indexOfLicense..indexOfLocation]);
+
+        if (indexOfLicense != -1)
+        {
+            lines.RemoveRange(indexOfLicense, indexOfLocation - indexOfLicense);
+        }
+
+        var linesDict = lines
             .Select(line => line.Split(':', 2))
             .Where(split => split.Length == 2)
             .Select(split => new KeyValuePair<string, string>(split[0].Trim(), split[1].Trim()))
@@ -38,22 +53,37 @@ public record PipShowResult
 
         return new PipShowResult
         {
-            Name = lines["Name"],
-            Version = lines["Version"],
-            Summary = lines.GetValueOrDefault("Summary"),
-            HomePage = lines.GetValueOrDefault("Home-page"),
-            Author = lines.GetValueOrDefault("Author"),
-            AuthorEmail = lines.GetValueOrDefault("Author-email"),
-            License = lines.GetValueOrDefault("License"),
-            Location = lines.GetValueOrDefault("Location"),
-            Requires = lines
+            Name = linesDict["Name"],
+            Version = linesDict["Version"],
+            Summary = linesDict.GetValueOrDefault("Summary"),
+            HomePage = linesDict.GetValueOrDefault("Home-page"),
+            Author = linesDict.GetValueOrDefault("Author"),
+            AuthorEmail = linesDict.GetValueOrDefault("Author-email"),
+            License = licenseText,
+            Location = linesDict.GetValueOrDefault("Location"),
+            Requires = linesDict
                 .GetValueOrDefault("Requires")
                 ?.Split(',', StringSplitOptions.TrimEntries)
                 .ToList(),
-            RequiredBy = lines
+            RequiredBy = linesDict
                 .GetValueOrDefault("Required-by")
                 ?.Split(',', StringSplitOptions.TrimEntries)
                 .ToList()
         };
     }
+
+    private static int GetIndexBySubstring(List<string> lines, string searchString)
+    {
+        var index = -1;
+        for (var i = 0; i < lines.Count; i++)
+        {
+            if (!lines[i].StartsWith(searchString, StringComparison.OrdinalIgnoreCase))
+                continue;
+
+            index = i;
+            break;
+        }
+
+        return index;
+    }
 }