cast solver order

Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2024-10-14 14:02:19 -04:00
parent 6bb688c371
commit 8047090fe8
+2 -2
View File
@@ -190,8 +190,8 @@ class DiffusionSampler:
self.config['thresholding'] = shared.opts.schedulers_use_thresholding
if 'lower_order_final' in self.config:
self.config['lower_order_final'] = shared.opts.schedulers_use_loworder
if 'solver_order' in self.config and shared.opts.schedulers_solver_order > 0:
self.config['solver_order'] = shared.opts.schedulers_solver_order
if 'solver_order' in self.config and int(shared.opts.schedulers_solver_order) > 0:
self.config['solver_order'] = (shared.opts.schedulers_solver_order)
if 'predict_x0' in self.config:
self.config['solver_type'] = shared.opts.uni_pc_variant
if 'beta_start' in self.config and shared.opts.schedulers_beta_start > 0: