diff --git a/StabilityMatrix/ViewModels/CheckpointBrowserViewModel.cs b/StabilityMatrix/ViewModels/CheckpointBrowserViewModel.cs index 9fe020cf..56b4ef1c 100644 --- a/StabilityMatrix/ViewModels/CheckpointBrowserViewModel.cs +++ b/StabilityMatrix/ViewModels/CheckpointBrowserViewModel.cs @@ -49,7 +49,11 @@ public partial class CheckpointBrowserViewModel : ObservableObject public IEnumerable AllCivitPeriods => Enum.GetValues(typeof(CivitPeriod)).Cast(); public IEnumerable AllSortModes => Enum.GetValues(typeof(CivitSortMode)).Cast(); - public IEnumerable AllModelTypes => Enum.GetValues(typeof(CivitModelType)).Cast(); + + public IEnumerable AllModelTypes => Enum.GetValues(typeof(CivitModelType)) + .Cast() + .Where(t => t != CivitModelType.AestheticGradient && t != CivitModelType.Poses) + .OrderBy(t => t.ToString()); public CheckpointBrowserViewModel( ICivitApi civitApi, @@ -67,7 +71,7 @@ public partial class CheckpointBrowserViewModel : ObservableObject ShowNsfw = settingsManager.Settings.ModelBrowserNsfwEnabled; SelectedPeriod = CivitPeriod.Month; SortMode = CivitSortMode.HighestRated; - SelectedModelType = CivitModelType.All; + SelectedModelType = CivitModelType.Checkpoint; HasSearched = false; CurrentPageNumber = 1; CanGoToPreviousPage = false;