Merge pull request #92 from ionite34/set-default-model-type

LykosAI · Jun 21, 2023 · fcaf8c9 · fcaf8c9
2 parents 4edf8f9 + 6f935d2
commit fcaf8c9
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/StabilityMatrix/ViewModels/CheckpointBrowserViewModel.cs b/StabilityMatrix/ViewModels/CheckpointBrowserViewModel.cs
@@ -49,7 +49,11 @@ public partial class CheckpointBrowserViewModel : ObservableObject
 
     public IEnumerable<CivitPeriod> AllCivitPeriods => Enum.GetValues(typeof(CivitPeriod)).Cast<CivitPeriod>();
     public IEnumerable<CivitSortMode> AllSortModes => Enum.GetValues(typeof(CivitSortMode)).Cast<CivitSortMode>();
-    public IEnumerable<CivitModelType> AllModelTypes => Enum.GetValues(typeof(CivitModelType)).Cast<CivitModelType>();
+
+    public IEnumerable<CivitModelType> AllModelTypes => Enum.GetValues(typeof(CivitModelType))
+        .Cast<CivitModelType>()
+        .Where(t => t != CivitModelType.AestheticGradient && t != CivitModelType.Poses)
+        .OrderBy(t => t.ToString());
 
     public CheckpointBrowserViewModel(
         ICivitApi civitApi, 
@@ -67,7 +71,7 @@ public CheckpointBrowserViewModel(
         ShowNsfw = settingsManager.Settings.ModelBrowserNsfwEnabled;
         SelectedPeriod = CivitPeriod.Month;
         SortMode = CivitSortMode.HighestRated;
-        SelectedModelType = CivitModelType.All;
+        SelectedModelType = CivitModelType.Checkpoint;
         HasSearched = false;
         CurrentPageNumber = 1;
         CanGoToPreviousPage = false;