multiple patches

This commit is contained in:
Vladimir Mandic
2024-05-27 16:29:32 -04:00
parent 2f323aa55d
commit b5c1ec1941
10 changed files with 40 additions and 17 deletions
+4 -1
View File
@@ -1,4 +1,5 @@
import os
import re
import inspect
from modules import shared
from modules import sd_samplers_common
@@ -128,7 +129,9 @@ class DiffusionSampler:
if shared.opts.schedulers_beta_schedule != 'default':
self.config['beta_schedule'] = shared.opts.schedulers_beta_schedule
if 'use_karras_sigmas' in self.config:
self.config['use_karras_sigmas'] = shared.opts.schedulers_use_karras
timesteps = re.split(',| ', shared.opts.schedulers_timesteps)
timesteps = [int(x) for x in timesteps if x.isdigit()]
self.config['use_karras_sigmas'] = shared.opts.schedulers_use_karras if len(timesteps) == 0 else False
if 'thresholding' in self.config:
self.config['thresholding'] = shared.opts.schedulers_use_thresholding
if 'lower_order_final' in self.config: