Merge pull request #365 from LykosAI/main

v2.7.6
LykosAI · Dec 28, 2023 · a9c48b8 · a9c48b8
2 parents 8ca421b + 136b7d8
commit a9c48b8
Show file tree

Hide file tree

Showing 12 changed files with 175 additions and 95 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -5,6 +5,17 @@ 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.7.6
+### Added
+- Added SDXL Turbo and Stable Video Diffusion to the Hugging Face tab
+### Changed
+- ControlNet model selector will now show the parent directory of a model when relevant
+### Fixed
+- Fixed Python Packages dialog crash due to pip commands including warnings
+- Fixed Base Model downloads from the Hugging Face tab downloading to the wrong folder
+- Fixed InvokeAI `! [rejected] v3.4.0post2 -> v3.4.0post2 (would clobber existing tag)` error on updating to the latest version 
+- Fixed settings not saving in some scenarios, such as when the `settings.json` file existed but was empty
+
 ## v2.7.5
 ### Fixed
 - Fixed Python Packages manager crash when pip list returns warnings in json

diff --git a/StabilityMatrix.Avalonia/Assets/hf-packages.json b/StabilityMatrix.Avalonia/Assets/hf-packages.json
@@ -38,6 +38,26 @@
     "LicenseType": "Open RAIL++",
     "LicensePath": "LICENSE.md"
   },
+  {
+    "ModelCategory": "BaseModel",
+    "ModelName": "SDXL Turbo",
+    "RepositoryPath": "stabilityai/sdxl-turbo",
+    "Files": [
+      "sd_xl_turbo_1.0_fp16.safetensors"
+    ],
+    "LicenseType": "SAI-NC-Community",
+    "LicensePath": "LICENSE.TXT"
+  },
+  {
+    "ModelCategory": "BaseModel",
+    "ModelName": "Stable Video Diffusion",
+    "RepositoryPath": "stabilityai/stable-video-diffusion-img2vid",
+    "Files": [
+      "svd.safetensors"
+    ],
+    "LicenseType": "Stable-Video-Diffusion-NC-Community",
+    "LicensePath": "LICENSE"
+  },
   {
     "ModelCategory": "ControlNet",
     "ModelName": "Canny",

diff --git a/StabilityMatrix.Avalonia/StabilityMatrix.Avalonia.csproj b/StabilityMatrix.Avalonia/StabilityMatrix.Avalonia.csproj
@@ -23,29 +23,29 @@
         <PackageReference Include="AsyncImageLoader.Avalonia" Version="3.2.1" />
         <PackageReference Include="AutoComplete.Net" Version="1.2211.2014.42"/>
         <PackageReference Include="Avalonia.AvaloniaEdit" Version="11.0.5" />
-        <PackageReference Include="Avalonia.Controls.PanAndZoom" Version="11.0.0" />
-        <PackageReference Include="Avalonia" Version="11.0.5" />
-        <PackageReference Include="Avalonia.Desktop" Version="11.0.5" />
-        <PackageReference Include="Avalonia.Fonts.Inter" Version="11.0.5" />
+        <PackageReference Include="Avalonia.Controls.PanAndZoom" Version="11.0.0.2" />
+        <PackageReference Include="Avalonia" Version="11.0.6" />
+        <PackageReference Include="Avalonia.Desktop" Version="11.0.6" />
+        <PackageReference Include="Avalonia.Fonts.Inter" Version="11.0.6" />
         <!--Condition below is needed to remove Avalonia.Diagnostics package from build output in Release configuration.-->
-        <PackageReference Condition="'$(Configuration)' == 'Debug'" Include="Avalonia.Diagnostics" Version="11.0.5" />
+        <PackageReference Condition="'$(Configuration)' == 'Debug'" Include="Avalonia.Diagnostics" Version="11.0.6" />
         <PackageReference Include="Avalonia.HtmlRenderer" Version="11.0.0" />
-        <PackageReference Include="Avalonia.Xaml.Behaviors" Version="11.0.5" />
+        <PackageReference Include="Avalonia.Xaml.Behaviors" Version="11.0.6" />
         <PackageReference Include="AvaloniaEdit.TextMate" Version="11.0.5" />
-        <PackageReference Include="bodong.Avalonia.PropertyGrid" Version="11.0.5.1" />
-        <PackageReference Include="bodong.PropertyModels" Version="11.0.5.1" />
+        <PackageReference Include="bodong.Avalonia.PropertyGrid" Version="11.0.6.1" />
+        <PackageReference Include="bodong.PropertyModels" Version="11.0.6.1" />
         <PackageReference Include="CommandLineParser" Version="2.9.1" />
         <PackageReference Include="CommunityToolkit.Mvvm" Version="8.2.2" />
         <PackageReference Include="DiscordRichPresence" Version="1.2.1.24" />
-        <PackageReference Include="Dock.Avalonia" Version="11.0.0.3" />
-        <PackageReference Include="Dock.Model.Avalonia" Version="11.0.0.3" />
-        <PackageReference Include="Dock.Serializer" Version="11.0.0.3" />
-        <PackageReference Include="DynamicData" Version="8.1.1" />
+        <PackageReference Include="Dock.Avalonia" Version="11.0.0.5" />
+        <PackageReference Include="Dock.Model.Avalonia" Version="11.0.0.5" />
+        <PackageReference Include="Dock.Serializer" Version="11.0.0.5" />
+        <PackageReference Include="DynamicData" Version="8.3.27" />
         <PackageReference Include="Exceptionless.DateTimeExtensions" Version="3.4.3" />
         <PackageReference Include="FluentAvalonia.BreadcrumbBar" Version="2.0.2" />
         <PackageReference Include="FluentAvaloniaUI" Version="2.0.4" />
-        <PackageReference Include="FluentIcons.Avalonia" Version="1.1.222" />
-        <PackageReference Include="FluentIcons.FluentAvalonia" Version="1.1.222" />
+        <PackageReference Include="FluentIcons.Avalonia" Version="1.1.224" />
+        <PackageReference Include="FluentIcons.FluentAvalonia" Version="1.1.224" />
         <PackageReference Include="FuzzySharp" Version="2.0.2" />
         <PackageReference Include="JetBrains.Annotations" Version="2023.3.0" />
         <PackageReference Include="Markdown.Avalonia" Version="11.0.2" />
@@ -59,16 +59,16 @@
         <PackageReference Include="Microsoft.Extensions.Http.Polly" Version="8.0.0" />
         <PackageReference Include="Microsoft.Extensions.Options.ConfigurationExtensions" Version="8.0.0" />
         <PackageReference Include="Nito.AsyncEx" Version="5.1.2" />
-        <PackageReference Include="NLog" Version="5.2.5" />
-        <PackageReference Include="NLog.Extensions.Logging" Version="5.3.5" />
+        <PackageReference Include="NLog" Version="5.2.7" />
+        <PackageReference Include="NLog.Extensions.Logging" Version="5.3.7" />
         <PackageReference Include="NSubstitute" Version="5.1.0" />
         <PackageReference Include="Polly" Version="8.2.0" />
         <PackageReference Include="Polly.Contrib.WaitAndRetry" Version="1.1.1" />
         <PackageReference Include="Polly.Extensions.Http" Version="3.0.0" />
-        <PackageReference Include="Projektanker.Icons.Avalonia.FontAwesome" Version="8.3.0" />
+        <PackageReference Include="Projektanker.Icons.Avalonia.FontAwesome" Version="9.0.1" />
         <PackageReference Include="RockLib.Reflection.Optimized" Version="2.0.0" />
-        <PackageReference Include="Sentry" Version="3.41.0" />
-        <PackageReference Include="Sentry.NLog" Version="3.41.0" />
+        <PackageReference Include="Sentry" Version="3.41.3" />
+        <PackageReference Include="Sentry.NLog" Version="3.41.3" />
         <PackageReference Include="SpacedGrid-Avalonia" Version="11.0.0" />
         <PackageReference Include="Sylvan.Common" Version="0.4.3" />
         <PackageReference Include="Sylvan.Data" Version="0.2.12" />

diff --git a/StabilityMatrix.Avalonia/ViewModels/CheckpointBrowser/HuggingFacePageViewModel.cs b/StabilityMatrix.Avalonia/ViewModels/CheckpointBrowser/HuggingFacePageViewModel.cs
@@ -41,7 +41,8 @@ public partial class HuggingFacePageViewModel : TabViewModelBase
     private readonly ISettingsManager settingsManager;
     private readonly INotificationService notificationService;
 
-    public SourceCache<HuggingfaceItem, string> ItemsCache { get; } = new(i => i.RepositoryPath + i.ModelName);
+    public SourceCache<HuggingfaceItem, string> ItemsCache { get; } =
+        new(i => i.RepositoryPath + i.ModelName);
 
     public IObservableCollection<CategoryViewModel> Categories { get; set; } =
         new ObservableCollectionExtended<CategoryViewModel>();
@@ -76,7 +77,13 @@ INotificationService notificationService
             .Connect()
             .DeferUntilLoaded()
             .Group(i => i.ModelCategory)
-            .Transform(g => new CategoryViewModel(g.Cache.Items) { Title = g.Key.GetDescription() ?? g.Key.ToString() })
+            .Transform(
+                g =>
+                    new CategoryViewModel(g.Cache.Items)
+                    {
+                        Title = g.Key.GetDescription() ?? g.Key.ToString()
+                    }
+            )
             .SortBy(vm => vm.Title)
             .Bind(Categories)
             .WhenAnyPropertyChanged()
@@ -137,12 +144,13 @@ private async Task ImportSelected()
         {
             foreach (var file in viewModel.Item.Files)
             {
-                var url = $"https://huggingface.co/{viewModel.Item.RepositoryPath}/resolve/main/{file}?download=true";
+                var url =
+                    $"https://huggingface.co/{viewModel.Item.RepositoryPath}/resolve/main/{file}?download=true";
                 var sharedFolderType = viewModel.Item.ModelCategory.ConvertTo<SharedFolderType>();
                 var downloadPath = new FilePath(
                     Path.Combine(
                         settingsManager.ModelsDirectory,
-                        sharedFolderType.GetDescription() ?? sharedFolderType.ToString(),
+                        sharedFolderType.ToString(),
                         viewModel.Item.Subfolder ?? string.Empty,
                         file
                     )

diff --git a/StabilityMatrix.Avalonia/ViewModels/Inference/Modules/ControlNetModule.cs b/StabilityMatrix.Avalonia/ViewModels/Inference/Modules/ControlNetModule.cs
@@ -42,9 +42,8 @@ protected override void OnApplyStep(ModuleApplyStepEventArgs e)
             {
                 Name = e.Nodes.GetUniqueName("ControlNet_LoadImage"),
                 Image =
-                    card.SelectImageCardViewModel.ImageSource?.GetHashGuidFileNameCached(
-                        "Inference"
-                    ) ?? throw new ValidationException("No ImageSource")
+                    card.SelectImageCardViewModel.ImageSource?.GetHashGuidFileNameCached("Inference")
+                    ?? throw new ValidationException("No ImageSource")
             }
         );
 
@@ -53,8 +52,7 @@ protected override void OnApplyStep(ModuleApplyStepEventArgs e)
             {
                 Name = e.Nodes.GetUniqueName("ControlNetLoader"),
                 ControlNetName =
-                    card.SelectedModel?.FileName
-                    ?? throw new ValidationException("No SelectedModel"),
+                    card.SelectedModel?.RelativePath ?? throw new ValidationException("No SelectedModel"),
             }
         );
 
@@ -64,10 +62,8 @@ protected override void OnApplyStep(ModuleApplyStepEventArgs e)
                 Name = e.Nodes.GetUniqueName("ControlNetApply"),
                 Image = imageLoad.Output1,
                 ControlNet = controlNetLoader.Output,
-                Positive =
-                    e.Temp.Conditioning?.Positive ?? throw new ArgumentException("No Conditioning"),
-                Negative =
-                    e.Temp.Conditioning?.Negative ?? throw new ArgumentException("No Conditioning"),
+                Positive = e.Temp.Conditioning?.Positive ?? throw new ArgumentException("No Conditioning"),
+                Negative = e.Temp.Conditioning?.Negative ?? throw new ArgumentException("No Conditioning"),
                 Strength = card.Strength,
                 StartPercent = card.StartPercent,
                 EndPercent = card.EndPercent,
@@ -93,10 +89,7 @@ protected override void OnApplyStep(ModuleApplyStepEventArgs e)
                 }
             );
 
-            e.Temp.RefinerConditioning = (
-                controlNetRefinerApply.Output1,
-                controlNetRefinerApply.Output2
-            );
+            e.Temp.RefinerConditioning = (controlNetRefinerApply.Output1, controlNetRefinerApply.Output2);
         }
     }
 }
diff --git a/StabilityMatrix.Avalonia/ViewModels/Inference/SelectImageCardViewModel.cs b/StabilityMatrix.Avalonia/ViewModels/Inference/SelectImageCardViewModel.cs
@@ -71,7 +71,9 @@ public void ApplyStep(ModuleApplyStepEventArgs e)
     {
         e.Builder.SetupImagePrimarySource(
             ImageSource ?? throw new ValidationException("Input Image is required"),
-            !CurrentBitmapSize.IsEmpty ? CurrentBitmapSize : throw new ValidationException("CurrentBitmapSize is null"),
+            !CurrentBitmapSize.IsEmpty
+                ? CurrentBitmapSize
+                : throw new ValidationException("CurrentBitmapSize is null"),
             e.Builder.Connections.BatchIndex
         );
     }
@@ -98,7 +100,10 @@ partial void OnImageSourceChanged(ImageSource? value)
     private async Task SelectImageFromFilePickerAsync()
     {
         var files = await App.StorageProvider.OpenFilePickerAsync(
-            new FilePickerOpenOptions { FileTypeFilter = [FilePickerFileTypes.ImagePng, FilePickerFileTypes.ImageJpg] }
+            new FilePickerOpenOptions
+            {
+                FileTypeFilter = [FilePickerFileTypes.ImagePng, FilePickerFileTypes.ImageJpg]
+            }
         );
 
         if (files.FirstOrDefault()?.TryGetLocalPath() is { } path)
@@ -112,7 +117,10 @@ private async Task SelectImageFromFilePickerAsync()
     /// </summary>
     public void DragOver(object? sender, DragEventArgs e)
     {
-        if (e.Data.GetDataFormats().Contains(DataFormats.Files) || e.Data.GetContext<LocalImageFile>() is not null)
+        if (
+            e.Data.GetDataFormats().Contains(DataFormats.Files)
+            || e.Data.GetContext<LocalImageFile>() is not null
+        )
         {
             e.Handled = true;
             return;
@@ -134,7 +142,10 @@ public void Drop(object? sender, DragEventArgs e)
             return;
         }
         // 2. OS Files
-        if (e.Data.GetFiles() is { } files && files.Select(f => f.TryGetLocalPath()).FirstOrDefault() is { } path)
+        if (
+            e.Data.GetFiles() is { } files
+            && files.Select(f => f.TryGetLocalPath()).FirstOrDefault() is { } path
+        )
         {
             e.Handled = true;
 
@@ -154,7 +165,7 @@ private void LoadUserImageSafe(ImageSource image)
         catch (Exception e)
         {
             Logger.Warn(e, "Error loading image");
-            notificationService.ShowPersistent("Error loading image", e.Message);
+            notificationService.Show("Error loading image", e.Message);
         }
     }
 

diff --git a/StabilityMatrix.Avalonia/Views/MainWindow.axaml.cs b/StabilityMatrix.Avalonia/Views/MainWindow.axaml.cs
@@ -132,15 +132,16 @@ protected override void OnLoaded(RoutedEventArgs e)
             return;
 
         // Navigate to first page
-        Dispatcher
-            .UIThread
-            .Post(
-                () =>
-                    navigationService.NavigateTo(
-                        vm.Pages[0],
-                        new BetterSlideNavigationTransition { Effect = SlideNavigationTransitionEffect.FromBottom }
-                    )
-            );
+        Dispatcher.UIThread.Post(
+            () =>
+                navigationService.NavigateTo(
+                    vm.Pages[0],
+                    new BetterSlideNavigationTransition
+                    {
+                        Effect = SlideNavigationTransitionEffect.FromBottom
+                    }
+                )
+        );
 
         // Check show update teaching tip
         if (vm.UpdateViewModel.IsUpdateAvailable)
@@ -165,27 +166,24 @@ private void NavigationService_OnTypedNavigation(object? sender, TypedNavigation
         var mainViewModel = (MainWindowViewModel)DataContext!;
 
         mainViewModel.SelectedCategory = mainViewModel
-            .Pages
-            .Concat(mainViewModel.FooterPages)
+            .Pages.Concat(mainViewModel.FooterPages)
             .FirstOrDefault(x => x.GetType() == e.ViewModelType);
     }
 
     private void OnUpdateAvailable(object? sender, UpdateInfo? updateInfo)
     {
-        Dispatcher
-            .UIThread
-            .Post(() =>
+        Dispatcher.UIThread.Post(() =>
+        {
+            if (DataContext is MainWindowViewModel vm && vm.ShouldShowUpdateAvailableTeachingTip(updateInfo))
             {
-                if (DataContext is MainWindowViewModel vm && vm.ShouldShowUpdateAvailableTeachingTip(updateInfo))
-                {
-                    var target = this.FindControl<NavigationViewItem>("FooterUpdateItem")!;
-                    var tip = this.FindControl<TeachingTip>("UpdateAvailableTeachingTip")!;
+                var target = this.FindControl<NavigationViewItem>("FooterUpdateItem")!;
+                var tip = this.FindControl<TeachingTip>("UpdateAvailableTeachingTip")!;
 
-                    tip.Target = target;
-                    tip.Subtitle = $"{Compat.AppVersion.ToDisplayString()} -> {updateInfo.Version}";
-                    tip.IsOpen = true;
-                }
-            });
+                tip.Target = target;
+                tip.Subtitle = $"{Compat.AppVersion.ToDisplayString()} -> {updateInfo.Version}";
+                tip.IsOpen = true;
+            }
+        });
     }
 
     public void SetDefaultFonts()
@@ -282,18 +280,16 @@ private void OnActualThemeVariantChanged(object? sender, EventArgs e)
 
     private void OnImageLoadFailed(object? sender, ImageLoadFailedEventArgs e)
     {
-        Dispatcher
-            .UIThread
-            .Post(() =>
-            {
-                var fileName = Path.GetFileName(e.Url);
-                var displayName = string.IsNullOrEmpty(fileName) ? e.Url : fileName;
-                notificationService.ShowPersistent(
-                    "Failed to load image",
-                    $"Could not load '{displayName}'\n({e.Exception.Message})",
-                    NotificationType.Warning
-                );
-            });
+        Dispatcher.UIThread.Post(() =>
+        {
+            var fileName = Path.GetFileName(e.Url);
+            var displayName = string.IsNullOrEmpty(fileName) ? e.Url : fileName;
+            notificationService.Show(
+                "Failed to load image",
+                $"Could not load '{displayName}'\n({e.Exception.Message})",
+                NotificationType.Warning
+            );
+        });
     }
 
     private void TryEnableMicaEffect()
@@ -319,7 +315,11 @@ private void TryEnableMicaEffect()
         else if (ActualThemeVariant == ThemeVariant.Light)
         {
             // Similar effect here
-            var color = this.TryFindResource("SolidBackgroundFillColorBase", ThemeVariant.Light, out var value)
+            var color = this.TryFindResource(
+                "SolidBackgroundFillColorBase",
+                ThemeVariant.Light,
+                out var value
+            )
                 ? (Color2)(Color)value!
                 : new Color2(243, 243, 243);