fix cn steps

Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-05-04 15:04:20 -04:00
parent 7aa09957dc
commit 5971f67e4f
3 changed files with 11 additions and 1 deletions
+1 -1
View File
@@ -460,7 +460,7 @@ def calculate_base_steps(p, use_denoise_start, use_refiner_start):
steps = p.steps
elif 'HiDreamImageEditingPipeline' in shared.sd_model.__class__.__name__:
steps = p.steps
elif shared.sd_model_type == 'omnigen':
elif shared.sd_model.__class__.__name__ in sd_models.i2i_pipes:
steps = p.steps
elif p.denoising_strength > 0:
steps = (p.steps // p.denoising_strength) + 1