v2.16.5

CHANGELOG.md

@@ -5,6 +5,44 @@ 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.16.5
+> Full technical notes for this release: [docs.lykos.ai/stability-matrix/release-notes/2.16.5](https://docs.lykos.ai/stability-matrix/release-notes/2.16.5)
+### Added
+- Added **Fizgig**, a LoRA training studio for Flux 2 Klein 9B, Krea 2, MiniMax H3 and Qwen Image 2.1 with dataset prep, training, LoRA repair and extraction in one app ([#1761](https://github.com/LykosAI/StabilityMatrix/pull/1761)) - thanks to @e-nord!
+- Added AMD ROCm support for **Stable Diffusion WebUI Forge - Neo** on Windows and Linux. Forge Neo's developer supports NVIDIA only, so report AMD issues with it to Stability Matrix ([#1770](https://github.com/LykosAI/StabilityMatrix/pull/1770)) - thanks to @NeuralFault!
+### Changed
+- ComfyUI now installs and updates from its current GitHub home, [Comfy-Org/ComfyUI](https://github.com/Comfy-Org/ComfyUI)
+### Fixed
+- Fixed **Wan2GP** installs on Python 3.11 failing with `sageattention ... has no wheels with a matching Python version tag`. New Wan2GP installs now use Python 3.11 with PyTorch 2.10 and CUDA 13 with matching Triton, SageAttention and Flash Attention, the same setup Wan2GP's own installer uses. CUDA 13 needs NVIDIA driver 580 or newer. GTX 10XX cards stay on Python 3.10
+- Existing **Wan2GP** installs on Python 3.10 keep their PyTorch 2.7 setup when updated. If a newer model such as LTX 2.5 fails with a Triton error, add Wan2GP again as a new install, which now uses Python 3.11, and move your outputs across ([#1756](https://github.com/LykosAI/StabilityMatrix/issues/1756))
+- Fixed ComfyUI updates downgrading SageAttention post6 to post4 and reinstalling compatible wheels ([#1764](https://github.com/LykosAI/StabilityMatrix/issues/1764)).
+- Fixed ComfyUI installation and updates failing on Python 3.13+ because the NumPy 1.x constraint has no compatible wheel ([#1754](https://github.com/LykosAI/StabilityMatrix/issues/1754))
+- Fixed **AI-Toolkit** installation failing on Linux while unpacking Node.js with a "Dangerous link path" error ([#1767](https://github.com/LykosAI/StabilityMatrix/issues/1767))
+- Fixed **Windows ROCm** PyTorch installs pairing mismatched `torch` and `torchvision` versions, which broke packages at `import torchvision` ([#1769](https://github.com/LykosAI/StabilityMatrix/pull/1769)) - thanks to @NeuralFault!
+- ControlNet preprocessing in Inference now offers to update older tagged Inference Core Nodes installs to a version with the ControlNet import fixes and Python 3.12/3.13 support
+- Fixed the package uninstall confirmation counting your whole shared model library for packages using symlink sharing. Only the package's own files are counted; shared models were never deleted
+- Fixed logs ignoring `--home-dir`, and opening a missing folder or file showing an error dialog instead of a short notice ([#1766](https://github.com/LykosAI/StabilityMatrix/pull/1766)) - thanks to @e-nord!
+- Fixed shared model folder links hiding **DiffusionModels** in the Checkpoint Manager, Image Lab and Inference, including when DiffusionModels is itself a link ([#1743](https://github.com/LykosAI/StabilityMatrix/pull/1743), [#1763](https://github.com/LykosAI/StabilityMatrix/pull/1763)) - thanks to @NeuralFault and @ionite34!
+- Fixed the Model Browser's **Base Model** filter being lost when using **Next Model** / **Previous Model** or refreshing a model's details ([#1436](https://github.com/LykosAI/StabilityMatrix/issues/1436))
+- Fixed **Open on CivArchive** opening a file URL instead of the model page on Linux ([#1765](https://github.com/LykosAI/StabilityMatrix/issues/1765))
+- Fixed emoji showing as `��` in Image Lab messages and in CivitAI and CivArchive model descriptions
+- Fixed release-note text being cut off beside the scrollbar in the update dialog ([#1757](https://github.com/LykosAI/StabilityMatrix/issues/1757))
+- Fixed text in **Image Lab** messages and thinking blocks being cut off at the right edge. Message bubbles now fit their text, and inline code is easier to read
+- Fixed **Save & Regenerate** in Image Lab deleting later images and dropping the edited message's attachments. It now shows progress and can be cancelled
+- Fixed **Regenerate** in Image Lab removing the old response before the new one was ready. A failed or cancelled regenerate now keeps it
+- Fixed Image Lab pulling you back to a conversation when its generation finished while you were in another, and showing that generation's error in the one you had open
+- Fixed deleting one image from a multi-image message in Image Lab deleting your original file when Image Lab hadn't kept its own copy
+- Fixed your message in Image Lab not being editable or deletable after a failed or cancelled generation
+- Fixed cancelling a Gemini generation in Image Lab showing "Generation failed" instead of "Cancelled"
+- Fixed Image Lab not scrolling to a new message when you were already at the bottom of the conversation
+- Fixed a ComfyUI generation in Image Lab not stopping when ComfyUI disconnected, if you had opened a Gemini conversation in the meantime
+- Fixed package version lists and update checks showing an old release as the newest, because startup update checks ran into GitHub's rate limit
+### Supporters
+#### 🌟 Visionaries
+This patch is a good one for AMD owners: Forge Neo now runs on ROCm on both Windows and Linux, and Windows ROCm installs always get a `torch` and `torchvision` that actually match. It also adds Fizgig, a whole LoRA training studio, to the package list. Most of it arrived as community contributions, and our Visionaries are the reason we have time to review that work properly and say yes to it. Thank you **Waterclouds**, **MrMxyzptlk12836**, **bluepopsicle**, **dispenser**, **Droolguy**, **TwistedDragon**, **snotty**, **cusalapapen1481**, **moon_milky2843**, **sn3232323233350**, **CC**, **hasezou11013179**, and **Akros**. It's great to see **SnooSnooEternal** back with us and **jweg79** moving up from the Pioneers, and hello to **japublicspot78364519** and **jonny90**, our newest Visionaries; we're so happy you're here. 💛
+#### 🚀 Pioneers
+To our Pioneers **SinthCore**, **Jisuren**, **Hurbie53**, **Cjloha**, **Alligator1907**, **CommissarGiygas16050**, **Zombop**, **Silerae**, **joshsciascia72**, **cy_hart691842**, and **okachako**: thank you for backing a patch full of fixes that make things work on more machines. And hi **acleveranagram**, our newest Pioneer; we're really glad to have you! 💛
+
 ## v2.16.4
 ### Added
 - Added **Comfy Kitchen Attention** (`--use-ck-attention`) as a **Cross Attention Method** launch option for ComfyUI and ComfyUI-Zluda - thanks to @e-nord!

StabilityMatrix.Avalonia/Controls/MarkdownHtmlLabel.cs

@@ -0,0 +1,54 @@
+using System;
+using Avalonia;
+using TheArtOfDev.HtmlRenderer.Avalonia;
+
+namespace StabilityMatrix.Avalonia.Controls;
+
+/// <summary>
+/// The HTML label behind <see cref="MarkdownViewer"/>. It fills the width it is offered, or with
+/// <see cref="FitToContent"/> takes only the width its text needs, wrapping at the width offered.
+/// </summary>
+public class MarkdownHtmlLabel : HtmlLabel
+{
+    public static readonly StyledProperty<bool> FitToContentProperty = AvaloniaProperty.Register<
+        MarkdownHtmlLabel,
+        bool
+    >(nameof(FitToContent));
+
+    static MarkdownHtmlLabel()
+    {
+        AffectsMeasure<MarkdownHtmlLabel>(FitToContentProperty);
+    }
+
+    /// <summary>
+    /// Size the width to the text, up to the width offered, instead of filling it.
+    /// </summary>
+    public bool FitToContent
+    {
+        get => GetValue(FitToContentProperty);
+        set => SetValue(FitToContentProperty, value);
+    }
+
+    protected override Size MeasureOverride(Size constraint)
+    {
+        if (!FitToContent || double.IsPositiveInfinity(constraint.Width))
+            return base.MeasureOverride(constraint);
+
+        var horizontal = Padding.Left + Padding.Right + BorderThickness.Left + BorderThickness.Right;
+        var vertical = Padding.Top + Padding.Bottom + BorderThickness.Top + BorderThickness.Bottom;
+        var available = Math.Max(constraint.Width - horizontal, 0);
+
+        // Lay out at the text's natural width, and wrap at the offered width only if it is wider
+        _htmlContainer.MaxSize = default;
+        _htmlContainer.PerformLayout();
+        if (_htmlContainer.ActualSize.Width > available)
+        {
+            _htmlContainer.MaxSize = new Size(available, 0);
+            _htmlContainer.PerformLayout();
+        }
+
+        // Rounded up so layout rounding can't hand back less than the text was laid out in
+        var size = _htmlContainer.ActualSize;
+        return new Size(Math.Ceiling(size.Width) + horizontal, Math.Ceiling(size.Height) + vertical);
+    }
+}

StabilityMatrix.Avalonia/Controls/MarkdownViewer.axaml

@@ -1,7 +1,6 @@
 <Styles
     xmlns="https://github.com/avaloniaui"
     xmlns:x="http://schemas.microsoft.com/winfx/2006/xaml"
-    xmlns:avalonia="clr-namespace:TheArtOfDev.HtmlRenderer.Avalonia;assembly=Avalonia.HtmlRenderer"
     xmlns:controls="using:StabilityMatrix.Avalonia.Controls">
     <Design.PreviewWith>
         <controls:MarkdownViewer
@@ -15,10 +14,11 @@
         <!--  Set Defaults  -->
         <Setter Property="Template">
             <ControlTemplate>
-                <avalonia:HtmlPanel
-                    x:Name="PART_HtmlPanel"
+                <controls:MarkdownHtmlLabel
+                    x:Name="PART_HtmlLabel"
+                    AutoSizeHeightOnly="True"
                     Background="Transparent"
-                    Text="{TemplateBinding Html}" />
+                    FitToContent="{TemplateBinding FitToContent}" />
             </ControlTemplate>
         </Setter>
     </Style>

StabilityMatrix.Avalonia/Controls/MarkdownViewer.axaml.cs

@@ -1,26 +1,36 @@
 using System.IO;
+using System.Text;
 using Avalonia;
 using Avalonia.Controls.Primitives;
 using Markdig;
 using TheArtOfDev.HtmlRenderer.Avalonia;
 
 namespace StabilityMatrix.Avalonia.Controls;
 
+/// <summary>
+/// Renders Markdown, or HTML set through the Html property, with the app's markdown stylesheet.
+/// </summary>
 public class MarkdownViewer : TemplatedControlBase
 {
+    private static readonly MarkdownPipeline Pipeline = new MarkdownPipelineBuilder()
+        .UseAdvancedExtensions()
+        .UseEmojiAndSmiley()
+        .Build();
+
+    private HtmlLabel? htmlLabel;
+
     public static readonly StyledProperty<string> TextProperty = AvaloniaProperty.Register<
         MarkdownViewer,
         string
     >(nameof(Text));
 
+    /// <summary>
+    /// Markdown to render.
+    /// </summary>
     public string Text
     {
         get => GetValue(TextProperty);
-        set
-        {
-            SetValue(TextProperty, value);
-            ParseText(value);
-        }
+        set => SetValue(TextProperty, value);
     }
 
     public static readonly StyledProperty<string> HtmlProperty = AvaloniaProperty.Register<
@@ -34,45 +44,88 @@ private string Html
         set => SetValue(HtmlProperty, value);
     }
 
+    public static readonly StyledProperty<bool> FitToContentProperty =
+        MarkdownHtmlLabel.FitToContentProperty.AddOwner<MarkdownViewer>();
+
+    /// <summary>
+    /// Size the width to the text, up to the width offered, instead of filling it. For chat
+    /// bubbles; leave it off where the content should span its container.
+    /// </summary>
+    public bool FitToContent
+    {
+        get => GetValue(FitToContentProperty);
+        set => SetValue(FitToContentProperty, value);
+    }
+
     public static readonly StyledProperty<string> CustomCssProperty = AvaloniaProperty.Register<
         MarkdownViewer,
         string
     >(nameof(CustomCss));
 
+    /// <summary>
+    /// CSS appended to the markdown stylesheet when the template is applied.
+    /// </summary>
     public string CustomCss
     {
         get => GetValue(CustomCssProperty);
         set => SetValue(CustomCssProperty, value);
     }
 
+    /// <summary>
+    /// Encodes every character outside the Basic Multilingual Plane (emoji, mostly) as a numeric
+    /// character reference. HtmlRenderer splits words at each UTF-16 code unit in U+4E00..U+FA2D,
+    /// a range that covers the surrogates, so a raw emoji is drawn as two U+FFFD halves; a
+    /// reference is decoded after that split and keeps the pair together.
+    /// </summary>
+    internal static string? EncodeAstralCharacters(string? html)
+    {
+        if (string.IsNullOrEmpty(html) || html.AsSpan().IndexOfAnyInRange('\uD800', '\uDFFF') < 0)
+            return html;
+
+        var builder = new StringBuilder(html.Length + 16);
+        foreach (var rune in html.EnumerateRunes())
+        {
+            if (rune.IsBmp)
+                builder.Append((char)rune.Value);
+            else
+                builder.Append("&#x").Append(rune.Value.ToString("X")).Append(';');
+        }
+
+        return builder.ToString();
+    }
+
     private void ParseText(string value)
     {
         if (string.IsNullOrWhiteSpace(value))
             return;
 
-        var pipeline = new MarkdownPipelineBuilder().UseAdvancedExtensions().UseEmojiAndSmiley().Build();
         var html =
-            $"""<html><body class="markdown-body">{Markdig.Markdown.ToHtml(value, pipeline)}</body></html>""";
+            $"""<html><body class="markdown-body">{Markdig.Markdown.ToHtml(value, Pipeline)}</body></html>""";
         Html = html;
     }
 
     protected override void OnApplyTemplate(TemplateAppliedEventArgs e)
     {
         base.OnApplyTemplate(e);
 
-        if (e.NameScope.Find("PART_HtmlPanel") is not HtmlPanel htmlPanel)
+        htmlLabel = e.NameScope.Find("PART_HtmlLabel") as HtmlLabel;
+        if (htmlLabel is null)
             return;
 
         using var cssFile = Assets.MarkdownCss.Open();
         using var reader = new StreamReader(cssFile);
         var css = reader.ReadToEnd();
 
-        htmlPanel.BaseStylesheet = $"{css}\n{CustomCss}";
+        htmlLabel.BaseStylesheet = $"{css}\n{CustomCss}";
 
         if (string.IsNullOrWhiteSpace(Html) && !string.IsNullOrWhiteSpace(Text))
         {
             ParseText(Text);
         }
+        else
+        {
+            htmlLabel.Text = EncodeAstralCharacters(Html);
+        }
     }
 
     protected override void OnPropertyChanged(AvaloniaPropertyChangedEventArgs change)
@@ -81,6 +134,10 @@ protected override void OnPropertyChanged(AvaloniaPropertyChangedEventArgs chang
         {
             ParseText(change.NewValue.ToString());
         }
+        else if (change.Property == HtmlProperty && htmlLabel is not null)
+        {
+            htmlLabel.Text = EncodeAstralCharacters(change.NewValue as string);
+        }
 
         base.OnPropertyChanged(change);
     }

StabilityMatrix.Avalonia/Controls/ReleaseNotesMarkdownViewer.cs

@@ -0,0 +1,24 @@
+using System.Linq;
+using Avalonia;
+using Avalonia.Controls;
+using Avalonia.VisualTree;
+
+namespace StabilityMatrix.Avalonia.Controls;
+
+/// <summary>
+/// Displays release notes with space between the document and the vertical scrollbar.
+/// </summary>
+public class ReleaseNotesMarkdownViewer : BetterMarkdownScrollViewer
+{
+    public ReleaseNotesMarkdownViewer()
+    {
+        if (
+            this.GetVisualChildren().OfType<ScrollViewer>().FirstOrDefault() is
+            { Content: Control content } viewer
+        )
+        {
+            viewer.Content = null;
+            viewer.Content = new Border { Child = content, Padding = new Thickness(0, 0, 18, 0) };
+        }
+    }
+}

StabilityMatrix.Avalonia/DialogHelper.cs

@@ -696,25 +696,14 @@ private static (int FileCount, long TotalBytes) GetFileStats(DirectoryInfo direc
         var fileCount = 0;
         var totalBytes = 0L;
 
-        foreach (var file in directory.EnumerateFiles())
+        foreach (
+            var file in directory.EnumerateFiles("*", EnumerationOptionConstants.AllDirectoriesExceptLinks)
+        )
         {
-            if (file.Attributes.HasFlag(FileAttributes.ReparsePoint))
-                continue;
-
             fileCount++;
             totalBytes += file.Length;
         }
 
-        foreach (var subDir in directory.EnumerateDirectories())
-        {
-            if (subDir.Attributes.HasFlag(FileAttributes.ReparsePoint))
-                continue;
-
-            var (subCount, subBytes) = GetFileStats(subDir);
-            fileCount += subCount;
-            totalBytes += subBytes;
-        }
-
         return (fileCount, totalBytes);
     }
 }

StabilityMatrix.Avalonia/Helpers/UnixPrerequisiteHelper.cs

@@ -560,7 +560,7 @@ public async Task InstallNodeIfNecessary(IProgress<ProgressReport>? progress = n
         );
 
         // unzip
-        await ArchiveHelper.Extract7ZAuto(nodeDownloadPath, AssetsDir);
+        await TarGZipExtractor.ExtractAsync(nodeDownloadPath, AssetsDir);
 
         var nodeDir = Compat.IsMacOS
             ? AssetsDir.JoinDir("node-v20.19.3-darwin-arm64")

StabilityMatrix.Avalonia/Models/BananaVision/Messages.cs

@@ -69,7 +69,9 @@ public ThinkingMessage(string thinkingContent)
     /// Preview of the thinking content (first ~100 chars)
     /// </summary>
     public string ThinkingPreview =>
-        ThinkingContent.Length > 100 ? ThinkingContent[..100] + "..." : ThinkingContent;
+        ThinkingContent.Length > 100
+            ? ThinkingContent[..(char.IsHighSurrogate(ThinkingContent[99]) ? 99 : 100)] + "..."
+            : ThinkingContent;
 
     /// <summary>
     /// Whether the thinking section is expanded