Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 01:04:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

cleanup

Browse Source 
Signed-off-by: vladmandic <mandic00@live.com>
This commit is contained in:
vladmandic
2026-03-13 11:49:41 +01:00
parent f91fd63a60
commit f6ed1a92a4
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/sd_samplers.py
+2 -2
View File
@@ -90,11 +90,11 @@ def create_sampler(name, model, scheduler_overrides=None):
if config is None or config.constructor is None:
return restore_default(model)
from modules import sd_samplers_diffusers
sd_samplers_diffusers._scheduler_overrides = scheduler_overrides or {}
sd_samplers_diffusers.scheduler_overrides = scheduler_overrides or {}
try:
sampler = config.constructor(model)
finally:
sd_samplers_diffusers._scheduler_overrides = {}
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')