From 35feb15ea6cc4ac32cab7aedf25c22dc5cd9019a Mon Sep 17 00:00:00 2001
From: x0x0b <42596409+x0x0b@users.noreply.github.com>
Date: Mon, 23 Feb 2026 15:14:01 +0900
Subject: [PATCH 1/4] feat: add Early Access-only update badge and preserve
 state on refresh

---
 .../Languages/Resources.Designer.cs           |  9 ++
 .../Languages/Resources.resx                  |  3 +
 StabilityMatrix.Avalonia/Styles/Card.axaml    |  9 ++
 .../Styles/ThemeColors.axaml                  | 92 ++++++++++---------
 .../CheckpointFileViewModel.cs                |  4 +
 .../Views/CheckpointsPage.axaml               | 20 +++-
 .../Models/Database/LocalModelFile.cs         | 20 +++-
 .../Services/ModelIndexService.cs             | 34 +++++++
 8 files changed, 141 insertions(+), 50 deletions(-)

diff --git a/StabilityMatrix.Avalonia/Languages/Resources.Designer.cs b/StabilityMatrix.Avalonia/Languages/Resources.Designer.cs
index 45706973..af726ddd 100644
--- a/StabilityMatrix.Avalonia/Languages/Resources.Designer.cs
+++ b/StabilityMatrix.Avalonia/Languages/Resources.Designer.cs
@@ -3304,6 +3304,15 @@ public static string Label_UpdateAvailable {
                 return ResourceManager.GetString("Label_UpdateAvailable", resourceCulture);
             }
         }
+
+        /// <summary>
+        ///   Looks up a localized string similar to Early Access Update Available.
+        /// </summary>
+        public static string Label_EarlyAccessUpdateAvailable {
+            get {
+                return ResourceManager.GetString("Label_EarlyAccessUpdateAvailable", resourceCulture);
+            }
+        }
         
         /// <summary>
         ///   Looks up a localized string similar to Updates.
diff --git a/StabilityMatrix.Avalonia/Languages/Resources.resx b/StabilityMatrix.Avalonia/Languages/Resources.resx
index 2d83d3c5..42d057cf 100644
--- a/StabilityMatrix.Avalonia/Languages/Resources.resx
+++ b/StabilityMatrix.Avalonia/Languages/Resources.resx
@@ -231,6 +231,9 @@
   <data name="Label_UpdateAvailable" xml:space="preserve">
     <value>Update Available</value>
   </data>
+  <data name="Label_EarlyAccessUpdateAvailable" xml:space="preserve">
+    <value>Early Access Update Available</value>
+  </data>
   <data name="Label_BecomeAPatron" xml:space="preserve">
     <value>Become a Patron</value>
   </data>
diff --git a/StabilityMatrix.Avalonia/Styles/Card.axaml b/StabilityMatrix.Avalonia/Styles/Card.axaml
index 1632eaf1..c2d40935 100644
--- a/StabilityMatrix.Avalonia/Styles/Card.axaml
+++ b/StabilityMatrix.Avalonia/Styles/Card.axaml
@@ -94,6 +94,15 @@
                 </Style>
             </Style>
 
+            <!--  Early Access  -->
+            <Style Selector="controls|Card.early-access">
+                <Setter Property="Background" Value="{DynamicResource ThemeDeepOrangeColorTransparent}" />
+                <Setter Property="BorderBrush" Value="{DynamicResource ThemeDeepOrangeColorTransparent}" />
+                <Style Selector="^ /template/ ContentPresenter#PART_ContentPresenter">
+                    <Setter Property="Foreground" Value="{DynamicResource ButtonForeground}" />
+                </Style>
+            </Style>
+
             <!--  Info  -->
             <Style Selector="controls|Card.info">
                 <Setter Property="Background" Value="{DynamicResource ThemeDarkBlueColorTransparent}" />
diff --git a/StabilityMatrix.Avalonia/Styles/ThemeColors.axaml b/StabilityMatrix.Avalonia/Styles/ThemeColors.axaml
index 963fb0f7..b53c1948 100644
--- a/StabilityMatrix.Avalonia/Styles/ThemeColors.axaml
+++ b/StabilityMatrix.Avalonia/Styles/ThemeColors.axaml
@@ -1,7 +1,8 @@
-<ResourceDictionary xmlns="https://github.com/avaloniaui"
-                    xmlns:x="http://schemas.microsoft.com/winfx/2006/xaml"
-                    xmlns:sty="clr-namespace:FluentAvalonia.Styling;assembly=FluentAvalonia">
-    <!-- Static colors -->
+<ResourceDictionary
+    xmlns="https://github.com/avaloniaui"
+    xmlns:x="http://schemas.microsoft.com/winfx/2006/xaml"
+    xmlns:sty="clr-namespace:FluentAvalonia.Styling;assembly=FluentAvalonia">
+    <!--  Static colors  -->
     <Color x:Key="ThemePrimaryColor">#333333</Color>
     <Color x:Key="ThemeDarkDarkRedColor">#952923</Color>
     <Color x:Key="ThemeDarkRedColor">#C2362E</Color>
@@ -32,58 +33,59 @@
     <Color x:Key="ThemeAmberColor">#FFC107</Color>
     <Color x:Key="ThemeOrangeColor">#FF9800</Color>
     <Color x:Key="ThemeDeepOrangeColor">#FF5722</Color>
+    <Color x:Key="ThemeDeepOrangeColorTransparent">#AAFF5722</Color>
     <Color x:Key="ThemeEldenRingOrangeColor">#FF4F00</Color>
     <Color x:Key="ThemeBrownColor">#795548</Color>
     <Color x:Key="ThemeGreyColor">#9E9E9E</Color>
     <Color x:Key="ThemeLightGreyColor">#C0C0C0</Color>
     <Color x:Key="ThemeBlueGreyColor">#607D8B</Color>
-    
-    <!-- Dark / light dynamic colors -->
+
+    <!--  Dark / light dynamic colors  -->
     <ResourceDictionary.ThemeDictionaries>
         <ResourceDictionary x:Key="Default">
-            <SolidColorBrush x:Key="ThemeBackgroundBrush" Color="#AAAAAA"/>
-            <SolidColorBrush x:Key="ThemeDarkAccentForegroundBrush" Color="#AAAAAA"/>
-            <SolidColorBrush x:Key="ThemeDarkAccentBackgroundBrush" Color="#686868"/>
-            <SolidColorBrush x:Key="RegionColor" Color="#AAAAAA"/>
-            
-            <SolidColorBrush x:Key="CompletionBorderBrush" Color="#43454A"/>
-            <SolidColorBrush x:Key="CompletionBackgroundBrush" Color="#2B2D30"/>
-            <SolidColorBrush x:Key="CompletionSecondaryBackgroundBrush" Color="#393B40"/>
-            <SolidColorBrush x:Key="CompletionForegroundBrush" Color="#B4B8BF"/>
-            <SolidColorBrush x:Key="CompletionSecondaryForegroundBrush" Color="#878B8D"/>
-            <SolidColorBrush x:Key="CompletionSelectionBackgroundBrush" Color="#2E436E"/>
-            <SolidColorBrush x:Key="CompletionSelectionForegroundBrush" Color="#5389F4"/>
+            <SolidColorBrush x:Key="ThemeBackgroundBrush" Color="#AAAAAA" />
+            <SolidColorBrush x:Key="ThemeDarkAccentForegroundBrush" Color="#AAAAAA" />
+            <SolidColorBrush x:Key="ThemeDarkAccentBackgroundBrush" Color="#686868" />
+            <SolidColorBrush x:Key="RegionColor" Color="#AAAAAA" />
+
+            <SolidColorBrush x:Key="CompletionBorderBrush" Color="#43454A" />
+            <SolidColorBrush x:Key="CompletionBackgroundBrush" Color="#2B2D30" />
+            <SolidColorBrush x:Key="CompletionSecondaryBackgroundBrush" Color="#393B40" />
+            <SolidColorBrush x:Key="CompletionForegroundBrush" Color="#B4B8BF" />
+            <SolidColorBrush x:Key="CompletionSecondaryForegroundBrush" Color="#878B8D" />
+            <SolidColorBrush x:Key="CompletionSelectionBackgroundBrush" Color="#2E436E" />
+            <SolidColorBrush x:Key="CompletionSelectionForegroundBrush" Color="#5389F4" />
         </ResourceDictionary>
-        
+
         <ResourceDictionary x:Key="Dark">
-            <SolidColorBrush x:Key="ThemeBackgroundBrush" Color="#282828"/>
-            <SolidColorBrush x:Key="ThemeDarkAccentForegroundBrush" Color="#AAAAAA"/>
-            <SolidColorBrush x:Key="ThemeDarkAccentBackgroundBrush" Color="#252525"/>
-            <SolidColorBrush x:Key="RegionColor" Color="#282828"/>
-            
-            <SolidColorBrush x:Key="CompletionBorderBrush" Color="#43454A"/>
-            <SolidColorBrush x:Key="CompletionBackgroundBrush" Color="#2B2D30"/>
-            <SolidColorBrush x:Key="CompletionSecondaryBackgroundBrush" Color="#393B40"/>
-            <SolidColorBrush x:Key="CompletionForegroundBrush" Color="#B4B8BF"/>
-            <SolidColorBrush x:Key="CompletionSecondaryForegroundBrush" Color="#878B8D"/>
-            <SolidColorBrush x:Key="CompletionSelectionBackgroundBrush" Color="#2E436E"/>
-            <SolidColorBrush x:Key="CompletionSelectionForegroundBrush" Color="#5389F4"/>
+            <SolidColorBrush x:Key="ThemeBackgroundBrush" Color="#282828" />
+            <SolidColorBrush x:Key="ThemeDarkAccentForegroundBrush" Color="#AAAAAA" />
+            <SolidColorBrush x:Key="ThemeDarkAccentBackgroundBrush" Color="#252525" />
+            <SolidColorBrush x:Key="RegionColor" Color="#282828" />
+
+            <SolidColorBrush x:Key="CompletionBorderBrush" Color="#43454A" />
+            <SolidColorBrush x:Key="CompletionBackgroundBrush" Color="#2B2D30" />
+            <SolidColorBrush x:Key="CompletionSecondaryBackgroundBrush" Color="#393B40" />
+            <SolidColorBrush x:Key="CompletionForegroundBrush" Color="#B4B8BF" />
+            <SolidColorBrush x:Key="CompletionSecondaryForegroundBrush" Color="#878B8D" />
+            <SolidColorBrush x:Key="CompletionSelectionBackgroundBrush" Color="#2E436E" />
+            <SolidColorBrush x:Key="CompletionSelectionForegroundBrush" Color="#5389F4" />
         </ResourceDictionary>
-        
+
         <ResourceDictionary x:Key="{x:Static sty:FluentAvaloniaTheme.HighContrastTheme}">
-            <SolidColorBrush x:Key="ThemeBackgroundBrush" Color="#AAAAAA"/>
-            <SolidColorBrush x:Key="ThemeDarkAccentForegroundBrush" Color="#AAAAAA"/>
-            <SolidColorBrush x:Key="ThemeDarkAccentBackgroundBrush" Color="#686868"/>
-            <SolidColorBrush x:Key="RegionColor" Color="#AAAAAA"/>
-            
-            <SolidColorBrush x:Key="CompletionBorderBrush" Color="#43454A"/>
-            <SolidColorBrush x:Key="CompletionBackgroundBrush" Color="#2B2D30"/>
-            <SolidColorBrush x:Key="CompletionSecondaryBackgroundBrush" Color="#393B40"/>
-            <SolidColorBrush x:Key="CompletionForegroundBrush" Color="#B4B8BF"/>
-            <SolidColorBrush x:Key="CompletionSecondaryForegroundBrush" Color="#878B8D"/>
-            <SolidColorBrush x:Key="CompletionSelectionBackgroundBrush" Color="#2E436E"/>
-            <SolidColorBrush x:Key="CompletionSelectionForegroundBrush" Color="#5389F4"/>
+            <SolidColorBrush x:Key="ThemeBackgroundBrush" Color="#AAAAAA" />
+            <SolidColorBrush x:Key="ThemeDarkAccentForegroundBrush" Color="#AAAAAA" />
+            <SolidColorBrush x:Key="ThemeDarkAccentBackgroundBrush" Color="#686868" />
+            <SolidColorBrush x:Key="RegionColor" Color="#AAAAAA" />
+
+            <SolidColorBrush x:Key="CompletionBorderBrush" Color="#43454A" />
+            <SolidColorBrush x:Key="CompletionBackgroundBrush" Color="#2B2D30" />
+            <SolidColorBrush x:Key="CompletionSecondaryBackgroundBrush" Color="#393B40" />
+            <SolidColorBrush x:Key="CompletionForegroundBrush" Color="#B4B8BF" />
+            <SolidColorBrush x:Key="CompletionSecondaryForegroundBrush" Color="#878B8D" />
+            <SolidColorBrush x:Key="CompletionSelectionBackgroundBrush" Color="#2E436E" />
+            <SolidColorBrush x:Key="CompletionSelectionForegroundBrush" Color="#5389F4" />
         </ResourceDictionary>
     </ResourceDictionary.ThemeDictionaries>
-    
+
 </ResourceDictionary>
diff --git a/StabilityMatrix.Avalonia/ViewModels/CheckpointManager/CheckpointFileViewModel.cs b/StabilityMatrix.Avalonia/ViewModels/CheckpointManager/CheckpointFileViewModel.cs
index eb8be6ab..251cf3bd 100644
--- a/StabilityMatrix.Avalonia/ViewModels/CheckpointManager/CheckpointFileViewModel.cs
+++ b/StabilityMatrix.Avalonia/ViewModels/CheckpointManager/CheckpointFileViewModel.cs
@@ -25,6 +25,8 @@ namespace StabilityMatrix.Avalonia.ViewModels.CheckpointManager;
 public partial class CheckpointFileViewModel : SelectableViewModelBase
 {
     [ObservableProperty]
+    [NotifyPropertyChangedFor(nameof(HasEarlyAccessUpdateOnly))]
+    [NotifyPropertyChangedFor(nameof(HasStandardUpdate))]
     private LocalModelFile checkpointFile;
 
     [ObservableProperty]
@@ -64,6 +66,8 @@ public partial class CheckpointFileViewModel : SelectableViewModelBase
     public bool CanShowTriggerWords => CheckpointFile.ConnectedModelInfo?.TrainedWords?.Length > 0;
     public string BaseModelName => CheckpointFile.ConnectedModelInfo?.BaseModel ?? string.Empty;
     public CivitModelType ModelType => CheckpointFile.ConnectedModelInfo?.ModelType ?? CivitModelType.Unknown;
+    public bool HasEarlyAccessUpdateOnly => CheckpointFile.HasEarlyAccessUpdateOnly;
+    public bool HasStandardUpdate => CheckpointFile.HasUpdate && !CheckpointFile.HasEarlyAccessUpdateOnly;
 
     /// <inheritdoc/>
     public CheckpointFileViewModel(
diff --git a/StabilityMatrix.Avalonia/Views/CheckpointsPage.axaml b/StabilityMatrix.Avalonia/Views/CheckpointsPage.axaml
index 58995bb6..ca9eabb3 100644
--- a/StabilityMatrix.Avalonia/Views/CheckpointsPage.axaml
+++ b/StabilityMatrix.Avalonia/Views/CheckpointsPage.axaml
@@ -629,7 +629,7 @@
                                             HorizontalAlignment="Left"
                                             VerticalAlignment="Top"
                                             Classes="success"
-                                            IsVisible="{Binding CheckpointFile.HasUpdate}">
+                                            IsVisible="{Binding HasStandardUpdate}">
 
                                             <TextBlock
                                                 HorizontalAlignment="Center"
@@ -640,6 +640,24 @@
                                                 <Run Text="{Binding CheckpointFile.LatestModelInfo.LatestVersionCreatedAt}" />
                                             </TextBlock>
                                         </controls:Card>
+                                        <controls:Card
+                                            Height="24"
+                                            Margin="4,8,0,0"
+                                            Padding="4"
+                                            HorizontalAlignment="Left"
+                                            VerticalAlignment="Top"
+                                            Classes="early-access"
+                                            IsVisible="{Binding HasEarlyAccessUpdateOnly}">
+
+                                            <TextBlock
+                                                HorizontalAlignment="Center"
+                                                VerticalAlignment="Center"
+                                                FontSize="11"
+                                                FontWeight="Medium">
+                                                <Run Text="{x:Static lang:Resources.Label_EarlyAccessUpdateAvailable}" />
+                                                <Run Text="{Binding CheckpointFile.LatestModelInfo.LatestVersionCreatedAt}" />
+                                            </TextBlock>
+                                        </controls:Card>
                                     </WrapPanel>
                                 </LayoutTransformControl>
 
diff --git a/StabilityMatrix.Core/Models/Database/LocalModelFile.cs b/StabilityMatrix.Core/Models/Database/LocalModelFile.cs
index 0e0581b9..23ee347a 100644
--- a/StabilityMatrix.Core/Models/Database/LocalModelFile.cs
+++ b/StabilityMatrix.Core/Models/Database/LocalModelFile.cs
@@ -24,12 +24,18 @@ public bool Equals(LocalModelFile? x, LocalModelFile? y)
                 return false;
             return x.RelativePath == y.RelativePath
                 && Equals(x.ConnectedModelInfo, y.ConnectedModelInfo)
-                && x.HasUpdate == y.HasUpdate;
+                && x.HasUpdate == y.HasUpdate
+                && x.HasEarlyAccessUpdateOnly == y.HasEarlyAccessUpdateOnly;
         }
 
         public int GetHashCode(LocalModelFile obj)
         {
-            return HashCode.Combine(obj.RelativePath, obj.ConnectedModelInfo, obj.HasUpdate);
+            return HashCode.Combine(
+                obj.RelativePath,
+                obj.ConnectedModelInfo,
+                obj.HasUpdate,
+                obj.HasEarlyAccessUpdateOnly
+            );
         }
     }
 
@@ -44,12 +50,13 @@ public virtual bool Equals(LocalModelFile? other)
             return true;
         return RelativePath == other.RelativePath
             && ConnectedModelInfo == other.ConnectedModelInfo
-            && HasUpdate == other.HasUpdate;
+            && HasUpdate == other.HasUpdate
+            && HasEarlyAccessUpdateOnly == other.HasEarlyAccessUpdateOnly;
     }
 
     public override int GetHashCode()
     {
-        return HashCode.Combine(RelativePath, ConnectedModelInfo, HasUpdate);
+        return HashCode.Combine(RelativePath, ConnectedModelInfo, HasUpdate, HasEarlyAccessUpdateOnly);
     }
 
     /// <summary>
@@ -88,6 +95,11 @@ public override int GetHashCode()
     /// </summary>
     public bool HasUpdate { get; set; }
 
+    /// <summary>
+    /// Whether updates are limited to Early Access versions only.
+    /// </summary>
+    public bool HasEarlyAccessUpdateOnly { get; set; }
+
     /// <summary>
     /// Last time this model was checked for an update
     /// </summary>
diff --git a/StabilityMatrix.Core/Services/ModelIndexService.cs b/StabilityMatrix.Core/Services/ModelIndexService.cs
index c6874d37..7db48e40 100644
--- a/StabilityMatrix.Core/Services/ModelIndexService.cs
+++ b/StabilityMatrix.Core/Services/ModelIndexService.cs
@@ -511,6 +511,7 @@ private async Task RefreshIndexParallelCore()
             if (modelsDict.TryGetValue(model.RelativePath, out var dbModel))
             {
                 model.HasUpdate = dbModel.HasUpdate;
+                model.HasEarlyAccessUpdateOnly = dbModel.HasEarlyAccessUpdateOnly;
                 model.LastUpdateCheck = dbModel.LastUpdateCheck;
                 model.LatestModelInfo = dbModel.LatestModelInfo;
             }
@@ -668,6 +669,7 @@ await liteDbContext.LocalModelFiles.FindAllAsync().ConfigureAwait(false) ?? []
                 .ToList();
 
             dbModel.HasUpdate = !latestHashes.Any(hash => installedHashes.Contains(hash!));
+            dbModel.HasEarlyAccessUpdateOnly = GetHasEarlyAccessUpdateOnly(dbModel, remoteModel);
             dbModel.LastUpdateCheck = DateTimeOffset.UtcNow;
             dbModel.LatestModelInfo = remoteModel;
 
@@ -700,4 +702,36 @@ private static HashSet<string> CollectModelHashes(IEnumerable<LocalModelFile> mo
         }
         return hashes;
     }
+
+    private static bool GetHasEarlyAccessUpdateOnly(LocalModelFile model, CivitModel? remoteModel)
+    {
+        if (!model.HasUpdate || !model.HasCivitMetadata)
+            return false;
+
+        var versions = remoteModel?.ModelVersions;
+        if (versions == null || versions.Count == 0)
+            return false;
+
+        var installedVersionId = model.ConnectedModelInfo?.VersionId;
+        if (installedVersionId == null)
+            return false;
+
+        var installedIndex = versions.FindIndex(version => version.Id == installedVersionId.Value);
+        if (installedIndex <= 0)
+            return false;
+
+        var hasEarlyAccess = false;
+        for (var i = 0; i < installedIndex; i++)
+        {
+            if (versions[i].IsEarlyAccess)
+            {
+                hasEarlyAccess = true;
+                continue;
+            }
+
+            return false;
+        }
+
+        return hasEarlyAccess;
+    }
 }

From fe91ee6b30d69d65d04a91d8fa9cb3cb31f8da22 Mon Sep 17 00:00:00 2001
From: x0x0b <42596409+x0x0b@users.noreply.github.com>
Date: Mon, 23 Feb 2026 15:32:02 +0900
Subject: [PATCH 2/4] feat: add Japanese localization for Early Access update
 label

---
 StabilityMatrix.Avalonia/Languages/Resources.ja-JP.resx | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/StabilityMatrix.Avalonia/Languages/Resources.ja-JP.resx b/StabilityMatrix.Avalonia/Languages/Resources.ja-JP.resx
index c416b6dc..2d935763 100644
--- a/StabilityMatrix.Avalonia/Languages/Resources.ja-JP.resx
+++ b/StabilityMatrix.Avalonia/Languages/Resources.ja-JP.resx
@@ -231,6 +231,9 @@
   <data name="Label_UpdateAvailable" xml:space="preserve">
     <value>更新あり</value>
   </data>
+  <data name="Label_EarlyAccessUpdateAvailable" xml:space="preserve">
+    <value>Early Accessの更新あり</value>
+  </data>
   <data name="Label_BecomeAPatron" xml:space="preserve">
     <value>Patreonになる</value>
   </data>
@@ -1432,4 +1435,4 @@ Sparkは兆レベルのパラメータを持つ基盤モデルで、膨大なパ
 ### 🔒 プライバシーファースト
 私たちはプライバシーを最優先に考えています。([生成AI利用規約](https://lykos.ai/gen-ai-terms)) **あなたのプロンプトや出力内容は、Lykos AIや必要なクラウドインフラパートナーによるAI学習には一切使用されません。** 処理はお客様の質問に対する返答を生成するためだけに安全を重視して使われ、**その後はプロンプト内容そのものではなく、メタデータ(タイムスタンプやトークン数など)のみを保存します。** あなたのデータが販売や共有されることは決してありません。</value>
   </data>
-</root>
\ No newline at end of file
+</root>

From bef42b54a54f36271b9237ec28859a10d0fc764f Mon Sep 17 00:00:00 2001
From: x0x0b <42596409+x0x0b@users.noreply.github.com>
Date: Mon, 23 Feb 2026 15:46:02 +0900
Subject: [PATCH 3/4] feat: add unit tests for Early Access update handling

---
 .../Avalonia/CheckpointFileViewModelTests.cs  |  86 +++++++++++
 .../Core/ModelIndexServiceTests.cs            | 135 ++++++++++++++++++
 .../Models/LocalModelFileTests.cs             |  61 ++++++++
 3 files changed, 282 insertions(+)
 create mode 100644 StabilityMatrix.Tests/Avalonia/CheckpointFileViewModelTests.cs
 create mode 100644 StabilityMatrix.Tests/Core/ModelIndexServiceTests.cs
 create mode 100644 StabilityMatrix.Tests/Models/LocalModelFileTests.cs

diff --git a/StabilityMatrix.Tests/Avalonia/CheckpointFileViewModelTests.cs b/StabilityMatrix.Tests/Avalonia/CheckpointFileViewModelTests.cs
new file mode 100644
index 00000000..c8f7c799
--- /dev/null
+++ b/StabilityMatrix.Tests/Avalonia/CheckpointFileViewModelTests.cs
@@ -0,0 +1,86 @@
+using Microsoft.Extensions.Logging;
+using NSubstitute;
+using StabilityMatrix.Avalonia.Services;
+using StabilityMatrix.Avalonia.ViewModels.Base;
+using StabilityMatrix.Avalonia.ViewModels.CheckpointManager;
+using StabilityMatrix.Core.Models;
+using StabilityMatrix.Core.Models.Api;
+using StabilityMatrix.Core.Models.Database;
+using StabilityMatrix.Core.Models.Settings;
+using StabilityMatrix.Core.Services;
+
+namespace StabilityMatrix.Tests.Avalonia;
+
+[TestClass]
+public class CheckpointFileViewModelTests
+{
+    [TestMethod]
+    public void HasStandardUpdate_IsFalse_WhenUpdateIsEarlyAccessOnly()
+    {
+        var vm = CreateViewModel(CreateCheckpointFile(hasUpdate: true, hasEarlyAccessUpdateOnly: true));
+
+        Assert.IsTrue(vm.HasEarlyAccessUpdateOnly);
+        Assert.IsFalse(vm.HasStandardUpdate);
+    }
+
+    [TestMethod]
+    public void CheckpointFile_Setter_RaisesDerivedUpdatePropertyNotifications()
+    {
+        var vm = CreateViewModel(CreateCheckpointFile(hasUpdate: true, hasEarlyAccessUpdateOnly: true));
+        var changed = new List<string>();
+
+        vm.PropertyChanged += (_, e) =>
+        {
+            if (!string.IsNullOrWhiteSpace(e.PropertyName))
+            {
+                changed.Add(e.PropertyName!);
+            }
+        };
+
+        vm.CheckpointFile = CreateCheckpointFile(hasUpdate: true, hasEarlyAccessUpdateOnly: false);
+
+        Assert.IsFalse(vm.HasEarlyAccessUpdateOnly);
+        Assert.IsTrue(vm.HasStandardUpdate);
+        CollectionAssert.Contains(changed, nameof(CheckpointFileViewModel.HasEarlyAccessUpdateOnly));
+        CollectionAssert.Contains(changed, nameof(CheckpointFileViewModel.HasStandardUpdate));
+    }
+
+    private static CheckpointFileViewModel CreateViewModel(LocalModelFile checkpointFile)
+    {
+        var settingsManager = Substitute.For<ISettingsManager>();
+        settingsManager.Settings.Returns(new Settings { ShowNsfwInCheckpointsPage = true });
+        settingsManager.IsLibraryDirSet.Returns(false);
+
+        return new CheckpointFileViewModel(
+            settingsManager,
+            Substitute.For<IModelIndexService>(),
+            Substitute.For<INotificationService>(),
+            Substitute.For<IDownloadService>(),
+            Substitute.For<IServiceManager<ViewModelBase>>(),
+            Substitute.For<ILogger>(),
+            checkpointFile
+        );
+    }
+
+    private static LocalModelFile CreateCheckpointFile(bool hasUpdate, bool hasEarlyAccessUpdateOnly)
+    {
+        return new LocalModelFile
+        {
+            RelativePath = "StableDiffusion/test-vm.safetensors",
+            SharedFolderType = SharedFolderType.StableDiffusion,
+            HasUpdate = hasUpdate,
+            HasEarlyAccessUpdateOnly = hasEarlyAccessUpdateOnly,
+            ConnectedModelInfo = new ConnectedModelInfo
+            {
+                ModelId = 77,
+                VersionId = 700,
+                Source = ConnectedModelSource.Civitai,
+                ModelName = "VM Test Model",
+                ModelDescription = string.Empty,
+                VersionName = "v700",
+                Tags = [],
+                Hashes = new CivitFileHashes(),
+            },
+        };
+    }
+}
diff --git a/StabilityMatrix.Tests/Core/ModelIndexServiceTests.cs b/StabilityMatrix.Tests/Core/ModelIndexServiceTests.cs
new file mode 100644
index 00000000..11053f24
--- /dev/null
+++ b/StabilityMatrix.Tests/Core/ModelIndexServiceTests.cs
@@ -0,0 +1,135 @@
+using System.Reflection;
+using StabilityMatrix.Core.Models;
+using StabilityMatrix.Core.Models.Api;
+using StabilityMatrix.Core.Models.Database;
+using StabilityMatrix.Core.Services;
+
+namespace StabilityMatrix.Tests.Core;
+
+[TestClass]
+public class ModelIndexServiceTests
+{
+    [TestMethod]
+    public void GetHasEarlyAccessUpdateOnly_ReturnsTrue_WhenAllNewerVersionsAreEarlyAccess()
+    {
+        var model = CreateLocalModel(installedVersionId: 100, hasUpdate: true);
+        var remoteModel = CreateRemoteModel(
+            CreateVersion(id: 300, isEarlyAccess: true),
+            CreateVersion(id: 200, isEarlyAccess: true),
+            CreateVersion(id: 100, isEarlyAccess: false)
+        );
+
+        var result = InvokeGetHasEarlyAccessUpdateOnly(model, remoteModel);
+
+        Assert.IsTrue(result);
+    }
+
+    [TestMethod]
+    public void GetHasEarlyAccessUpdateOnly_ReturnsFalse_WhenAnyNewerVersionIsPublic()
+    {
+        var model = CreateLocalModel(installedVersionId: 100, hasUpdate: true);
+        var remoteModel = CreateRemoteModel(
+            CreateVersion(id: 300, isEarlyAccess: true),
+            CreateVersion(id: 200, isEarlyAccess: false),
+            CreateVersion(id: 100, isEarlyAccess: false)
+        );
+
+        var result = InvokeGetHasEarlyAccessUpdateOnly(model, remoteModel);
+
+        Assert.IsFalse(result);
+    }
+
+    [TestMethod]
+    public void GetHasEarlyAccessUpdateOnly_ReturnsFalse_WhenInstalledVersionIsLatest()
+    {
+        var model = CreateLocalModel(installedVersionId: 100, hasUpdate: true);
+        var remoteModel = CreateRemoteModel(
+            CreateVersion(id: 100, isEarlyAccess: false),
+            CreateVersion(id: 90, isEarlyAccess: true)
+        );
+
+        var result = InvokeGetHasEarlyAccessUpdateOnly(model, remoteModel);
+
+        Assert.IsFalse(result);
+    }
+
+    [TestMethod]
+    public void GetHasEarlyAccessUpdateOnly_ReturnsFalse_WhenModelHasNoUpdate()
+    {
+        var model = CreateLocalModel(installedVersionId: 100, hasUpdate: false);
+        var remoteModel = CreateRemoteModel(
+            CreateVersion(id: 300, isEarlyAccess: true),
+            CreateVersion(id: 200, isEarlyAccess: true),
+            CreateVersion(id: 100, isEarlyAccess: false)
+        );
+
+        var result = InvokeGetHasEarlyAccessUpdateOnly(model, remoteModel);
+
+        Assert.IsFalse(result);
+    }
+
+    private static bool InvokeGetHasEarlyAccessUpdateOnly(LocalModelFile model, CivitModel? remoteModel)
+    {
+        var method = typeof(ModelIndexService).GetMethod(
+            "GetHasEarlyAccessUpdateOnly",
+            BindingFlags.NonPublic | BindingFlags.Static
+        );
+
+        Assert.IsNotNull(method);
+
+        var result = method.Invoke(null, [model, remoteModel]);
+
+        Assert.IsNotNull(result);
+
+        return (bool)result;
+    }
+
+    private static LocalModelFile CreateLocalModel(int installedVersionId, bool hasUpdate)
+    {
+        return new LocalModelFile
+        {
+            RelativePath = "StableDiffusion/test-model.safetensors",
+            SharedFolderType = SharedFolderType.StableDiffusion,
+            HasUpdate = hasUpdate,
+            ConnectedModelInfo = new ConnectedModelInfo
+            {
+                ModelId = 123,
+                VersionId = installedVersionId,
+                Source = ConnectedModelSource.Civitai,
+                ModelName = "Test Model",
+                ModelDescription = string.Empty,
+                VersionName = $"v{installedVersionId}",
+                Tags = [],
+                Hashes = new CivitFileHashes(),
+            },
+        };
+    }
+
+    private static CivitModel CreateRemoteModel(params CivitModelVersion[] versions)
+    {
+        return new CivitModel
+        {
+            Id = 123,
+            Name = "Test Model",
+            Description = string.Empty,
+            Type = CivitModelType.Unknown,
+            Tags = [],
+            Stats = new CivitModelStats(),
+            ModelVersions = versions.ToList(),
+        };
+    }
+
+    private static CivitModelVersion CreateVersion(int id, bool isEarlyAccess)
+    {
+        return new CivitModelVersion
+        {
+            Id = id,
+            Name = $"v{id}",
+            Description = string.Empty,
+            DownloadUrl = string.Empty,
+            TrainedWords = [],
+            Availability = isEarlyAccess ? "EarlyAccess" : "Public",
+            Stats = new CivitModelStats(),
+        };
+    }
+}
diff --git a/StabilityMatrix.Tests/Models/LocalModelFileTests.cs b/StabilityMatrix.Tests/Models/LocalModelFileTests.cs
new file mode 100644
index 00000000..4831989b
--- /dev/null
+++ b/StabilityMatrix.Tests/Models/LocalModelFileTests.cs
@@ -0,0 +1,61 @@
+using StabilityMatrix.Core.Models;
+using StabilityMatrix.Core.Models.Api;
+using StabilityMatrix.Core.Models.Database;
+
+namespace StabilityMatrix.Tests.Models;
+
+[TestClass]
+public class LocalModelFileTests
+{
+    [TestMethod]
+    public void Equals_ReturnsFalse_WhenEarlyAccessOnlyFlagDiffers()
+    {
+        var standardUpdateModel = CreateLocalModelFile(hasEarlyAccessUpdateOnly: false);
+        var earlyAccessOnlyModel = standardUpdateModel with { HasEarlyAccessUpdateOnly = true };
+
+        Assert.IsFalse(standardUpdateModel.Equals(earlyAccessOnlyModel));
+        Assert.IsFalse(
+            LocalModelFile.RelativePathConnectedModelInfoComparer.Equals(
+                standardUpdateModel,
+                earlyAccessOnlyModel
+            )
+        );
+    }
+
+    [TestMethod]
+    public void RelativePathConnectedModelInfoComparer_TreatsEarlyAccessFlagAsDistinct()
+    {
+        var standardUpdateModel = CreateLocalModelFile(hasEarlyAccessUpdateOnly: false);
+        var earlyAccessOnlyModel = standardUpdateModel with { HasEarlyAccessUpdateOnly = true };
+
+        var set = new HashSet<LocalModelFile>(LocalModelFile.RelativePathConnectedModelInfoComparer)
+        {
+            standardUpdateModel,
+            earlyAccessOnlyModel,
+        };
+
+        Assert.AreEqual(2, set.Count);
+    }
+
+    private static LocalModelFile CreateLocalModelFile(bool hasEarlyAccessUpdateOnly)
+    {
+        return new LocalModelFile
+        {
+            RelativePath = "StableDiffusion/model-a.safetensors",
+            SharedFolderType = SharedFolderType.StableDiffusion,
+            HasUpdate = true,
+            HasEarlyAccessUpdateOnly = hasEarlyAccessUpdateOnly,
+            ConnectedModelInfo = new ConnectedModelInfo
+            {
+                ModelId = 123,
+                VersionId = 101,
+                Source = ConnectedModelSource.Civitai,
+                ModelName = "Model A",
+                ModelDescription = string.Empty,
+                VersionName = "v101",
+                Tags = [],
+                Hashes = new CivitFileHashes(),
+            },
+        };
+    }
+}

From 14693ef8d58793b6d4f72f28e0da75365e4fe0b3 Mon Sep 17 00:00:00 2001
From: x0x0b <42596409+x0x0b@users.noreply.github.com>
Date: Mon, 23 Feb 2026 16:08:59 +0900
Subject: [PATCH 4/4] feat: simplify Early Access update check and add unit
 test

---
 .../Services/ModelIndexService.cs                 | 14 +-------------
 .../Core/ModelIndexServiceTests.cs                | 15 +++++++++++++++
 2 files changed, 16 insertions(+), 13 deletions(-)

diff --git a/StabilityMatrix.Core/Services/ModelIndexService.cs b/StabilityMatrix.Core/Services/ModelIndexService.cs
index 7db48e40..924642da 100644
--- a/StabilityMatrix.Core/Services/ModelIndexService.cs
+++ b/StabilityMatrix.Core/Services/ModelIndexService.cs
@@ -720,18 +720,6 @@ private static bool GetHasEarlyAccessUpdateOnly(LocalModelFile model, CivitModel
         if (installedIndex <= 0)
             return false;
 
-        var hasEarlyAccess = false;
-        for (var i = 0; i < installedIndex; i++)
-        {
-            if (versions[i].IsEarlyAccess)
-            {
-                hasEarlyAccess = true;
-                continue;
-            }
-
-            return false;
-        }
-
-        return hasEarlyAccess;
+        return versions.Take(installedIndex).All(version => version.IsEarlyAccess);
     }
 }
diff --git a/StabilityMatrix.Tests/Core/ModelIndexServiceTests.cs b/StabilityMatrix.Tests/Core/ModelIndexServiceTests.cs
index 11053f24..5dbd15de 100644
--- a/StabilityMatrix.Tests/Core/ModelIndexServiceTests.cs
+++ b/StabilityMatrix.Tests/Core/ModelIndexServiceTests.cs
@@ -68,6 +68,21 @@ public void GetHasEarlyAccessUpdateOnly_ReturnsFalse_WhenModelHasNoUpdate()
         Assert.IsFalse(result);
     }
 
+    [TestMethod]
+    public void GetHasEarlyAccessUpdateOnly_ReturnsFalse_WhenInstalledVersionIsNotInRemoteList()
+    {
+        var model = CreateLocalModel(installedVersionId: 100, hasUpdate: true);
+        var remoteModel = CreateRemoteModel(
+            CreateVersion(id: 300, isEarlyAccess: true),
+            CreateVersion(id: 200, isEarlyAccess: true),
+            CreateVersion(id: 150, isEarlyAccess: false)
+        );
+
+        var result = InvokeGetHasEarlyAccessUpdateOnly(model, remoteModel);
+
+        Assert.IsFalse(result);
+    }
+
     private static bool InvokeGetHasEarlyAccessUpdateOnly(LocalModelFile model, CivitModel? remoteModel)
     {
         var method = typeof(ModelIndexService).GetMethod(