Add inference workflow profiles

Author: mohnjiles

CHANGELOG.md

@@ -9,6 +9,7 @@ and this project adheres to [Semantic Versioning 2.0](https://semver.org/spec/v2
 ### Added
 - 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
 - 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
+- Added an Inference workflow profile selector with recommended defaults for Z-Image Base and Z-Image Turbo
 ### Changed
 - 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
 ### Fixed

StabilityMatrix.Avalonia/Controls/Inference/ModelCard.axaml

@@ -239,8 +239,46 @@
                                 </Grid>
                             </Expander.Header>
                             <StackPanel Margin="0,4,0,0" Spacing="8">
+                                <!--  Workflow Profile  -->
+                                <Grid ColumnDefinitions="90,*,Auto">
+                                    <TextBlock
+                                        Grid.Column="0"
+                                        VerticalAlignment="Center"
+                                        Text="Workflow"
+                                        TextAlignment="Left" />
+                                    <controls:BetterComboBox
+                                        Grid.Column="1"
+                                        Padding="8,6,4,6"
+                                        HorizontalAlignment="Stretch"
+                                        ItemsSource="{Binding WorkflowProfiles}"
+                                        SelectedItem="{Binding SelectedWorkflowProfile}">
+                                        <controls:BetterComboBox.Resources>
+                                            <converters:EnumStringConverter x:Key="EnumStringConverter" />
+                                        </controls:BetterComboBox.Resources>
+                                        <controls:BetterComboBox.ItemTemplate>
+                                            <DataTemplate>
+                                                <TextBlock Text="{Binding ., Converter={StaticResource EnumStringConverter}}" />
+                                            </DataTemplate>
+                                        </controls:BetterComboBox.ItemTemplate>
+                                        <controls:BetterComboBox.SelectionBoxItemTemplate>
+                                            <DataTemplate>
+                                                <TextBlock Text="{Binding ., Converter={StaticResource EnumStringConverter}}" />
+                                            </DataTemplate>
+                                        </controls:BetterComboBox.SelectionBoxItemTemplate>
+                                    </controls:BetterComboBox>
+                                    <Button
+                                        Grid.Column="2"
+                                        Margin="4,0,0,0"
+                                        VerticalAlignment="Stretch"
+                                        Command="{Binding ApplyRecommendedDefaultsCommand}"
+                                        IsEnabled="{Binding HasRecommendedDefaults}"
+                                        ToolTip.Tip="Apply recommended sampler defaults">
+                                        <fluent:SymbolIcon FontSize="16" Symbol="Sparkle" />
+                                    </Button>
+                                </Grid>
+
                                 <!--  Encoder Type  -->
-                                <Grid ColumnDefinitions="90,*">
+                                <Grid ColumnDefinitions="90,*" IsVisible="{Binding ShowEncoderTypeSelection}">
                                     <TextBlock
                                         Grid.Column="0"
                                         VerticalAlignment="Center"

StabilityMatrix.Avalonia/Models/Inference/InferenceWorkflowProfile.cs

@@ -0,0 +1,33 @@
+using StabilityMatrix.Core.Extensions;
+
+namespace StabilityMatrix.Avalonia.Models.Inference;
+
+public enum InferenceWorkflowProfile
+{
+    [StringValue("Auto")]
+    Auto,
+
+    [StringValue("Default / Checkpoint")]
+    DefaultCheckpoint,
+
+    [StringValue("Flux")]
+    Flux,
+
+    [StringValue("Flux.2")]
+    Flux2,
+
+    [StringValue("Z-Image Base")]
+    ZImageBase,
+
+    [StringValue("Z-Image Turbo")]
+    ZImageTurbo,
+
+    [StringValue("Anima / SD")]
+    Anima,
+
+    [StringValue("HiDream")]
+    HiDream,
+
+    [StringValue("Custom")]
+    Custom,
+}

StabilityMatrix.Avalonia/ViewModels/Inference/InferenceTextToImageViewModel.cs

@@ -14,6 +14,7 @@
 using StabilityMatrix.Core.Attributes;
 using StabilityMatrix.Core.Extensions;
 using StabilityMatrix.Core.Models;
+using StabilityMatrix.Core.Models.Api.Comfy;
 using StabilityMatrix.Core.Models.Api.Comfy.Nodes;
 using StabilityMatrix.Core.Models.Inference;
 using StabilityMatrix.Core.Models.Settings;
@@ -77,6 +78,7 @@ TabContext tabContext
         SeedCardViewModel.GenerateNewSeed();
 
         ModelCardViewModel = vmFactory.Get<ModelCardViewModel>();
+        ModelCardViewModel.RecommendedDefaultsRequested += ApplyRecommendedDefaults;
 
         // When the model changes in the ModelCardViewModel, we'll have access to it in the TabContext
 
@@ -169,7 +171,7 @@ protected override void BuildPrompt(BuildPromptEventArgs args)
 
         var useSd3Latent =
             SamplerCardViewModel.ModulesCardViewModel.IsModuleEnabled<FluxGuidanceModule>()
-            || (IsUnetLoader && !IsStableDiffusionUnet && !IsFlux2Unet);
+            || (IsUnetLoader && UsesSd3Latent);
         var usePlasmaNoise = SamplerCardViewModel.ModulesCardViewModel.IsModuleEnabled<PlasmaNoiseModule>();
 
         if (IsFlux2Unet)
@@ -245,16 +247,62 @@ protected override void BuildPrompt(BuildPromptEventArgs args)
 
     protected bool IsUnetLoader => ModelCardViewModel.SelectedModelLoader is ModelLoader.Unet;
 
-    protected bool IsStableDiffusionUnet =>
-        IsUnetLoader && ModelCardViewModel.SelectedClipType is "stable_diffusion";
+    protected InferenceWorkflowProfile ResolvedWorkflowProfile => ModelCardViewModel.ResolvedWorkflowProfile;
 
-    protected bool IsFlux2Unet => IsUnetLoader && ModelCardViewModel.SelectedClipType is "flux2";
+    protected bool IsAnimaUnet =>
+        IsUnetLoader
+        && (
+            ResolvedWorkflowProfile is InferenceWorkflowProfile.Anima
+            || (
+                ResolvedWorkflowProfile is InferenceWorkflowProfile.Custom
+                && ModelCardViewModel.SelectedClipType is "stable_diffusion"
+            )
+        );
+
+    protected bool IsFlux2Unet =>
+        IsUnetLoader
+        && (
+            ResolvedWorkflowProfile is InferenceWorkflowProfile.Flux2
+            || (
+                ResolvedWorkflowProfile is InferenceWorkflowProfile.Custom
+                && ModelCardViewModel.SelectedClipType is "flux2"
+            )
+        );
 
-    protected bool IsLumina2Unet => IsUnetLoader && ModelCardViewModel.SelectedClipType is "lumina2";
+    protected bool IsZImageUnet =>
+        IsUnetLoader
+        && (
+            ResolvedWorkflowProfile
+                is InferenceWorkflowProfile.ZImageBase
+                    or InferenceWorkflowProfile.ZImageTurbo
+            || (
+                ResolvedWorkflowProfile is InferenceWorkflowProfile.Custom
+                && ModelCardViewModel.SelectedClipType is "lumina2"
+            )
+        );
+
+    protected bool UsesSd3Latent =>
+        ResolvedWorkflowProfile
+            is InferenceWorkflowProfile.Flux
+                or InferenceWorkflowProfile.ZImageBase
+                or InferenceWorkflowProfile.ZImageTurbo
+                or InferenceWorkflowProfile.HiDream
+        || (
+            ResolvedWorkflowProfile is InferenceWorkflowProfile.Custom
+            && ModelCardViewModel.SelectedClipType is not "stable_diffusion" and not "flux2"
+        );
+
+    protected bool UsesFluxGuidanceSampler =>
+        ResolvedWorkflowProfile is InferenceWorkflowProfile.Flux or InferenceWorkflowProfile.HiDream
+        || (
+            ResolvedWorkflowProfile is InferenceWorkflowProfile.Custom
+            && IsUnetLoader
+            && ModelCardViewModel.SelectedClipType is not "stable_diffusion" and not "lumina2" and not "flux2"
+        );
 
     protected void ApplyModelSamplingForCurrentWorkflow(ModuleApplyStepEventArgs applyArgs)
     {
-        if (!IsLumina2Unet)
+        if (!IsZImageUnet)
             return;
 
         var builder = applyArgs.Builder;
@@ -281,7 +329,7 @@ protected void ApplySamplerForCurrentWorkflow(
         }
         else if (
             (includeGgufAsFluxGuidance && ModelCardViewModel.IsGguf)
-            || (IsUnetLoader && !IsStableDiffusionUnet && !IsLumina2Unet)
+            || (IsUnetLoader && UsesFluxGuidanceSampler)
         )
         {
             SamplerCardViewModel.ApplyStepsInitialCustomSampler(applyArgs, true);
@@ -296,6 +344,25 @@ protected void ApplySamplerForCurrentWorkflow(
         }
     }
 
+    private void ApplyRecommendedDefaults(InferenceWorkflowProfile profile)
+    {
+        switch (profile)
+        {
+            case InferenceWorkflowProfile.ZImageTurbo:
+                SamplerCardViewModel.SelectedSampler = ComfySampler.ResMultistep;
+                SamplerCardViewModel.SelectedScheduler = ComfyScheduler.Simple;
+                SamplerCardViewModel.Steps = 8;
+                SamplerCardViewModel.CfgScale = 1.0d;
+                break;
+            case InferenceWorkflowProfile.ZImageBase:
+                SamplerCardViewModel.SelectedSampler = ComfySampler.ResMultistep;
+                SamplerCardViewModel.SelectedScheduler = ComfyScheduler.Simple;
+                SamplerCardViewModel.Steps = 30;
+                SamplerCardViewModel.CfgScale = 4.0d;
+                break;
+        }
+    }
+
     /// <inheritdoc />
     protected override IEnumerable<ImageSource> GetInputImages()
     {