1580158
/
automatic
mirror of https://github.com/vladmandic/automatic
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

cleanup 

Signed-off-by: vladmandic <mandic00@live.com>
pull/4690/head
vladmandic 2026-03-13 11:49:41 +01:00
parent f91fd63a60
commit f6ed1a92a4
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
modules/sd_samplers.py
View File

@ -90,11 +90,11 @@ def create_sampler(name, model, scheduler_overrides=None):
if config is None or config.constructor is None: if config is None or config.constructor is None:
return restore_default(model) return restore_default(model)
from modules import sd_samplers_diffusers from modules import sd_samplers_diffusers
sd_samplers_diffusers._scheduler_overrides = scheduler_overrides or {} sd_samplers_diffusers.scheduler_overrides = scheduler_overrides or {}
try: try:
sampler = config.constructor(model) sampler = config.constructor(model)
finally: finally:
sd_samplers_diffusers._scheduler_overrides = {} sd_samplers_diffusers.scheduler_overrides = {}
if sampler.sampler is None: if sampler.sampler is None:
return restore_default(model) return restore_default(model)
is_flow = ('FlowMatch' in sampler.sampler.__class__.__name__) or (getattr(sampler.sampler.config, 'prediction_type', None) == 'flow_prediction') is_flow = ('FlowMatch' in sampler.sampler.__class__.__name__) or (getattr(sampler.sampler.config, 'prediction_type', None) == 'flow_prediction')