Skip to content

Commit 88d74ce

Browse files
committed
Add inference workflow recommended defaults
1 parent 57f80e6 commit 88d74ce

4 files changed

Lines changed: 32 additions & 4 deletions

File tree

CHANGELOG.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ and this project adheres to [Semantic Versioning 2.0](https://semver.org/spec/v2
99
### Added
1010
- Added support for the civitai.red (mature-content) domain — NSFW CivitAI links now open and copy as civitai.red URLs, and pasting a civitai.red URL into the CivitAI model browser search works the same as a civitai.com URL
1111
- Added official Inference support for Z-Image/Z-Image-Turbo, Anima, and Flux.2 UNet-only workflows, including their text encoder, latent, scheduler, and sampling requirements
12-
- Added an Inference workflow profile selector with recommended defaults for Z-Image Base and Z-Image Turbo
12+
- Added an Inference workflow profile selector with recommended defaults for default checkpoints, Flux, Flux.2, Z-Image Base/Turbo, and Anima
1313
### Changed
1414
- The CivitAI base model type filter now uses CivitAI's official `/api/v1/enums` endpoint, with fallbacks to the previous technique and a built-in list, so the filter stays populated even if the CivitAI response format changes or the service is unreachable
1515
### Fixed

StabilityMatrix.Avalonia/Controls/Inference/ModelCard.axaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -112,7 +112,7 @@
112112
</Grid>
113113

114114
<!-- Workflow Profile -->
115-
<Grid Margin="0,8,0,0" ColumnDefinitions="60,*,Auto">
115+
<Grid Margin="0,8,0,0" ColumnDefinitions="90,*,Auto">
116116
<TextBlock
117117
Grid.Column="0"
118118
VerticalAlignment="Center"

StabilityMatrix.Avalonia/ViewModels/Inference/InferenceTextToImageViewModel.cs

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -348,6 +348,24 @@ private void ApplyRecommendedDefaults(InferenceWorkflowProfile profile)
348348
{
349349
switch (profile)
350350
{
351+
case InferenceWorkflowProfile.DefaultCheckpoint:
352+
SamplerCardViewModel.SelectedSampler = ComfySampler.EulerAncestral;
353+
SamplerCardViewModel.SelectedScheduler = ComfyScheduler.Normal;
354+
SamplerCardViewModel.Steps = 30;
355+
SamplerCardViewModel.CfgScale = 5.0d;
356+
break;
357+
case InferenceWorkflowProfile.Flux:
358+
SamplerCardViewModel.SelectedSampler = ComfySampler.Euler;
359+
SamplerCardViewModel.SelectedScheduler = ComfyScheduler.Simple;
360+
SamplerCardViewModel.Steps = 20;
361+
SamplerCardViewModel.CfgScale = 3.5d;
362+
break;
363+
case InferenceWorkflowProfile.Flux2:
364+
SamplerCardViewModel.SelectedSampler = ComfySampler.Euler;
365+
SamplerCardViewModel.SelectedScheduler = ComfyScheduler.Normal;
366+
SamplerCardViewModel.Steps = 20;
367+
SamplerCardViewModel.CfgScale = 5.0d;
368+
break;
351369
case InferenceWorkflowProfile.ZImageTurbo:
352370
SamplerCardViewModel.SelectedSampler = ComfySampler.ResMultistep;
353371
SamplerCardViewModel.SelectedScheduler = ComfyScheduler.Simple;
@@ -360,6 +378,12 @@ private void ApplyRecommendedDefaults(InferenceWorkflowProfile profile)
360378
SamplerCardViewModel.Steps = 30;
361379
SamplerCardViewModel.CfgScale = 4.0d;
362380
break;
381+
case InferenceWorkflowProfile.Anima:
382+
SamplerCardViewModel.SelectedSampler = ComfySampler.ErSde;
383+
SamplerCardViewModel.SelectedScheduler = ComfyScheduler.Simple;
384+
SamplerCardViewModel.Steps = 30;
385+
SamplerCardViewModel.CfgScale = 4.0d;
386+
break;
363387
}
364388
}
365389

StabilityMatrix.Avalonia/ViewModels/Inference/ModelCardViewModel.cs

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -267,8 +267,12 @@ SelectedWorkflowProfile is InferenceWorkflowProfile.Auto
267267
: SelectedWorkflowProfile;
268268
public bool HasRecommendedDefaults =>
269269
ResolvedWorkflowProfile
270-
is InferenceWorkflowProfile.ZImageBase
271-
or InferenceWorkflowProfile.ZImageTurbo;
270+
is InferenceWorkflowProfile.DefaultCheckpoint
271+
or InferenceWorkflowProfile.Flux
272+
or InferenceWorkflowProfile.Flux2
273+
or InferenceWorkflowProfile.ZImageBase
274+
or InferenceWorkflowProfile.ZImageTurbo
275+
or InferenceWorkflowProfile.Anima;
272276

273277
public event Action<InferenceWorkflowProfile>? RecommendedDefaultsRequested;
274278

0 commit comments

Comments
 (0)