Add sigmoid beta scheduler

2026-09-19 17:24:32 +02:00 · 2025-05-04 17:42:22 +03:00
parent 0ef2249d55
commit f4dfe20bc1
3 changed files with 4 additions and 2 deletions
@@ -232,6 +232,8 @@ class DiffusionSampler:
                self.config['beta_schedule'] = 'scaled_linear'
            elif shared.opts.schedulers_beta_schedule == 'cosine':
                self.config['beta_schedule'] = 'squaredcos_cap_v2'
+            elif shared.opts.schedulers_beta_schedule == 'sigmoid':
+                self.config['beta_schedule'] = 'sigmoid'

        timesteps = re.split(',| ', shared.opts.schedulers_timesteps)
        timesteps = [int(x) for x in timesteps if x.isdigit()]