Fix hires overrides

12 months ago · f761927e03
1 changed files with 11 additions and 3 deletions
--- a/StabilityMatrix.Avalonia/ViewModels/Inference/Modules/ModuleBase.cs
+++ b/StabilityMatrix.Avalonia/ViewModels/Inference/Modules/ModuleBase.cs
@ -20,9 +20,17 @@ public abstract class ModuleBase : StackExpanderViewModel, IComfyStep, IInputIma
    /// <inheritdoc />
    public void ApplyStep(ModuleApplyStepEventArgs e)
    {
-        if (
-            (e.IsEnabledOverrides.TryGetValue(GetType(), out var isEnabledOverride) && !isEnabledOverride) || !IsEnabled
-        )
+        if (e.IsEnabledOverrides.TryGetValue(GetType(), out var isEnabledOverride))
+        {
+            if (isEnabledOverride)
+            {
+                OnApplyStep(e);
+            }
+
+            return;
+        }
+
+        if (!IsEnabled)
        {
            return;
        }