From f91fd63a60f9f381dc2eba0088afaa340cdd1732 Mon Sep 17 00:00:00 2001 From: vladmandic Date: Fri, 13 Mar 2026 11:43:30 +0100 Subject: [PATCH] merge: modules/sd_samplers.py --- modules/sd_samplers.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/modules/sd_samplers.py b/modules/sd_samplers.py index b113b65cc..1d47df5b4 100644 --- a/modules/sd_samplers.py +++ b/modules/sd_samplers.py @@ -66,7 +66,7 @@ def restore_default(model): return model.scheduler -def create_sampler(name, model): +def create_sampler(name, model, scheduler_overrides=None): if name is None or name == 'None': return model.scheduler if model is not None else None @@ -89,7 +89,12 @@ def create_sampler(name, model): config = find_sampler_config(name) if config is None or config.constructor is None: return restore_default(model) - sampler = config.constructor(model) + from modules import sd_samplers_diffusers + sd_samplers_diffusers._scheduler_overrides = scheduler_overrides or {} + try: + sampler = config.constructor(model) + finally: + sd_samplers_diffusers._scheduler_overrides = {} if sampler.sampler is None: return restore_default(model) is_flow = ('FlowMatch' in sampler.sampler.__class__.__name__) or (getattr(sampler.sampler.config, 'prediction_type', None) == 'flow_prediction')