Skip to content

Commit 9a71855

Browse files
mohnjilesmohnjiles
authored
Merge pull request #238 from ionite34/show-model-images
Add toggle for "Show Model Images" on checkpoints page
2 parents 1458797 + 633cf04 commit 9a71855

File tree

10 files changed

+601
-329
lines changed

10 files changed

+601
-329
lines changed

‎CHANGELOG.md

+3
Original file line numberDiff line numberDiff line change
@@ -8,11 +8,14 @@ and this project adheres to [Semantic Versioning 2.0](https://semver.org/spec/v2
88
## v2.4.0
99
### Added
1010
- New installable Package - [Fooocus-MRE](https://github.com/MoonRide303/Fooocus-MRE)
11+
- Added toggle to show connected model images in the Checkpoints tab
12+
- Added "Find Connected Metadata" option to the context menu of Checkpoint Folders in the Checkpoints tab to connect models that don't have any metadata
1113
### Changed
1214
- Revamped package installer
1315
- Added "advanced options" section for commit, shared folder method, and pytorch options
1416
- Can be run in the background
1517
- Shows progress in the Downloads tab
18+
- Even more performance improvements for loading and searching the Checkpoints page
1619
### Fixed
1720
- Fixed [#97](https://github.com/LykosAI/StabilityMatrix/issues/97) - Codeformer folder should now get linked correctly
1821
- Fixed [#106](https://github.com/LykosAI/StabilityMatrix/issues/106) - ComfyUI should now install correctly on Windows machines with an AMD GPU using DirectML

‎StabilityMatrix.Avalonia/DesignData/DesignData.cs

+51-44
Original file line numberDiff line numberDiff line change
@@ -165,67 +165,74 @@ public static void Initialize()
165165
InstallerViewModel.SelectedPackage = InstallerViewModel.AvailablePackages[0];
166166
InstallerViewModel.ReleaseNotes = "## Release Notes\nThis is a test release note.";
167167

168-
// Checkpoints page
169-
CheckpointsPageViewModel.CheckpointFolders = new ObservableCollection<CheckpointFolder>
170-
{
171-
new(settingsManager, downloadService, modelFinder)
168+
/*// Checkpoints page
169+
CheckpointsPageViewModel.CheckpointFolders =
170+
new CheckpointFolder[]
172171
{
173-
Title = "StableDiffusion",
174-
DirectoryPath = "Models/StableDiffusion",
175-
CheckpointFiles = new AdvancedObservableList<CheckpointFile>
172+
new(settingsManager, downloadService, modelFinder, notificationService)
176173
{
177-
new()
174+
Title = "StableDiffusion",
175+
DirectoryPath = "Models/StableDiffusion",
176+
CheckpointFiles = CheckpointFile[]
178177
{
179-
FilePath = "~/Models/StableDiffusion/electricity-light.safetensors",
180-
Title = "Auroral Background",
181-
PreviewImagePath =
182-
"https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/"
183-
+ "78fd2a0a-42b6-42b0-9815-81cb11bb3d05/00009-2423234823.jpeg",
184-
ConnectedModel = new ConnectedModelInfo
178+
new()
185179
{
186-
VersionName = "Lightning Auroral",
187-
BaseModel = "SD 1.5",
188-
ModelName = "Auroral Background",
189-
ModelType = CivitModelType.Model,
190-
FileMetadata = new CivitFileMetadata
180+
FilePath = "~/Models/StableDiffusion/electricity-light.safetensors",
181+
Title = "Auroral Background",
182+
PreviewImagePath =
183+
"https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/"
184+
+ "78fd2a0a-42b6-42b0-9815-81cb11bb3d05/00009-2423234823.jpeg",
185+
ConnectedModel = new ConnectedModelInfo
191186
{
192-
Format = CivitModelFormat.SafeTensor,
193-
Fp = CivitModelFpType.fp16,
194-
Size = CivitModelSize.pruned,
187+
VersionName = "Lightning Auroral",
188+
BaseModel = "SD 1.5",
189+
ModelName = "Auroral Background",
190+
ModelType = CivitModelType.Model,
191+
FileMetadata = new CivitFileMetadata
192+
{
193+
Format = CivitModelFormat.SafeTensor,
194+
Fp = CivitModelFpType.fp16,
195+
Size = CivitModelSize.pruned,
196+
}
195197
}
196-
}
198+
},
199+
new()
200+
{
201+
FilePath = "~/Models/Lora/model.safetensors",
202+
Title = "Some model"
203+
},
197204
},
198-
new() { FilePath = "~/Models/Lora/model.safetensors", Title = "Some model" },
199205
},
200-
},
201-
new(settingsManager, downloadService, modelFinder)
202-
{
203-
Title = "Lora",
204-
DirectoryPath = "Packages/Lora",
205-
SubFolders = new AdvancedObservableList<CheckpointFolder>()
206+
new(settingsManager, downloadService, modelFinder, notificationService)
206207
{
207-
new(settingsManager, downloadService, modelFinder)
208+
Title = "Lora",
209+
DirectoryPath = "Packages/Lora",
210+
SubFolders = CheckpointFolder[]
208211
{
209-
Title = "StableDiffusion",
210-
DirectoryPath = "Packages/Lora/Subfolder",
212+
new(settingsManager, downloadService, modelFinder, notificationService)
213+
{
214+
Title = "StableDiffusion",
215+
DirectoryPath = "Packages/Lora/Subfolder",
216+
},
217+
new(settingsManager, downloadService, modelFinder, notificationService)
218+
{
219+
Title = "Lora",
220+
DirectoryPath = "Packages/StableDiffusion/Subfolder",
221+
}
211222
},
212-
new(settingsManager, downloadService, modelFinder)
223+
CheckpointFiles = new AdvancedObservableList<CheckpointFile>
213224
{
214-
Title = "Lora",
215-
DirectoryPath = "Packages/StableDiffusion/Subfolder",
225+
new() { FilePath = "~/Models/Lora/lora_v2.pt", Title = "Best Lora v2", }
216226
}
217-
},
218-
CheckpointFiles = new AdvancedObservableList<CheckpointFile>
219-
{
220-
new() { FilePath = "~/Models/Lora/lora_v2.pt", Title = "Best Lora v2", }
221227
}
222-
}
223-
};
228+
};
224229
225230
foreach (var folder in CheckpointsPageViewModel.CheckpointFolders)
226231
{
227-
folder.DisplayedCheckpointFiles = folder.CheckpointFiles;
228-
}
232+
folder.DisplayedCheckpointFiles = new AdvancedObservableList<CheckpointFile>(
233+
folder.CheckpointFiles
234+
);
235+
}*/
229236

230237
CheckpointBrowserViewModel.ModelCards =
231238
new ObservableCollection<CheckpointBrowserCardViewModel>

‎StabilityMatrix.Avalonia/Languages/Resources.Designer.cs

+18
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

‎StabilityMatrix.Avalonia/Languages/Resources.resx

+6
Original file line numberDiff line numberDiff line change
@@ -246,4 +246,10 @@
246246
<data name="Label_LicenseAgreement" xml:space="preserve">
247247
<value>License Agreement.</value>
248248
</data>
249+
<data name="Label_FindConnectedMetadata" xml:space="preserve">
250+
<value>Find Connected Metadata</value>
251+
</data>
252+
<data name="Label_ShowModelImages" xml:space="preserve">
253+
<value>Show Model Images</value>
254+
</data>
249255
</root>

0 commit comments

Comments
 (0)