force update deis settings

This commit is contained in:
Vladimir Mandic
2023-10-20 09:31:51 -04:00
parent 3794af95c5
commit d8bb3496df
3 changed files with 7 additions and 3 deletions
+4 -2
View File
@@ -97,11 +97,13 @@ class DiffusionSampler:
self.config['solver_order'] = shared.opts.schedulers_solver_order
if 'predict_x0' in self.config:
self.config['predict_x0'] = shared.opts.uni_pc_variant
if name == 'DPM++ 2M':
self.config['algorithm_type'] = shared.opts.schedulers_dpm_solver
if 'beta_start' in self.config and shared.opts.schedulers_beta_start > 0:
self.config['beta_start'] = shared.opts.schedulers_beta_start
if 'beta_end' in self.config and shared.opts.schedulers_beta_end > 0:
self.config['beta_end'] = shared.opts.schedulers_beta_end
if name == 'DPM++ 2M':
self.config['algorithm_type'] = shared.opts.schedulers_dpm_solver
if name == 'DEIS':
self.config['algorithm_type'] = 'deis'
self.sampler = constructor(**self.config)
self.sampler.name = name