From 01db4d8140ac0f9d6b465cc9323fa938ae5127b2 Mon Sep 17 00:00:00 2001
From: vladmandic <mandic00@live.com>
Date: Thu, 22 Jan 2026 19:50:16 +0100
Subject: [PATCH] use refiner/detail steps as-is for non sd/sdxl models

Signed-off-by: vladmandic <mandic00@live.com>
---
 CHANGELOG.md                  |  1 +
 modules/processing_helpers.py | 18 ++++++++++--------
 2 files changed, 11 insertions(+), 8 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 42c29e497..4ae80780f 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -9,6 +9,7 @@
 - fix image delete  
 - fix `all_seeds` error  
 - improve `wrap_gradio` error handling  
+- use refiner/detail steps as-is for non sd/sdxl models  
 
 ## Update for 2026-01-20
 
diff --git a/modules/processing_helpers.py b/modules/processing_helpers.py
index 786156bf0..4bd7dd033 100644
--- a/modules/processing_helpers.py
+++ b/modules/processing_helpers.py
@@ -408,9 +408,11 @@ def calculate_base_steps(p, use_denoise_start, use_refiner_start):
 
 
 def calculate_hires_steps(p):
-    cls = shared.sd_model.__class__.__name__
-    if 'Flex' in cls or 'Kontext' in cls or 'Edit' in cls or 'Wan' in cls or 'Flux2' in cls:
-        steps = p.steps
+    if shared.sd_model_type not in ['sd', 'sdxl']:
+        if p.hr_second_pass_steps > 0:
+            steps = p.hr_second_pass_steps
+        else:
+            steps = p.steps
     elif p.hr_second_pass_steps > 0:
         steps = (p.hr_second_pass_steps // p.denoising_strength) + 1
     elif p.denoising_strength > 0:
@@ -422,10 +424,7 @@ def calculate_hires_steps(p):
 
 
 def calculate_refiner_steps(p):
-    cls = shared.sd_model.__class__.__name__
-    if 'Flex' in cls or 'Kontext' in cls or 'Edit' in cls or 'Wan' in cls or 'Flux2' in cls:
-        steps = p.steps
-    elif "StableDiffusionXL" in shared.sd_refiner.__class__.__name__:
+    if shared.sd_refiner_type == 'sdxl':
         if p.refiner_start > 0 and p.refiner_start < 1:
             steps = (p.refiner_steps // (1 - p.refiner_start) // 2) + 1
         elif p.denoising_strength > 0:
@@ -433,7 +432,10 @@ def calculate_refiner_steps(p):
         else:
             steps = 0
     else:
-        steps = (p.refiner_steps * 1.25) + 1
+        if p.refiner_steps > 0:
+            steps = p.refiner_steps
+        else:
+            steps = p.steps
     debug_steps(f'Steps: type=refiner input={p.refiner_steps} output={steps} start={p.refiner_start} denoise={p.denoising_strength}')
     return max(1, int(steps))