add flux2

Signed-off-by: vladmandic <mandic00@live.com>
This commit is contained in:
vladmandic
2025-11-26 15:47:17 +00:00
parent 70b96daa63
commit ad6458aa7b
14 changed files with 86 additions and 15 deletions
+1 -1
View File
@@ -408,7 +408,7 @@ def calculate_base_steps(p, use_denoise_start, use_refiner_start):
if len(getattr(p, 'timesteps', [])) > 0:
return None
cls = shared.sd_model.__class__.__name__
if 'Flex' in cls or 'Kontext' in cls or 'Edit' in cls or 'Wan' in cls:
if 'Flex' in cls or 'Kontext' in cls or 'Edit' in cls or 'Wan' in cls or 'Flux2':
steps = p.steps
elif is_modular():
steps = p.steps