From 96ae1ec0191ce91a1bb9c8c9b98f851f98b69d08 Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Mon, 18 Nov 2024 14:26:08 -0500 Subject: [PATCH] set default flowmatch shift value Signed-off-by: Vladimir Mandic --- modules/sd_samplers_diffusers.py | 10 ++++++++-- modules/shared.py | 2 +- 2 files changed, 9 insertions(+), 3 deletions(-) diff --git a/modules/sd_samplers_diffusers.py b/modules/sd_samplers_diffusers.py index afd3f4588..3cf153bf1 100644 --- a/modules/sd_samplers_diffusers.py +++ b/modules/sd_samplers_diffusers.py @@ -103,7 +103,7 @@ config = { 'LMSD': { 'use_karras_sigmas': False, 'use_exponential_sigmas': False, 'use_beta_sigmas': False, 'timestep_spacing': 'linspace', 'steps_offset': 0 }, 'KDPM2': { 'use_karras_sigmas': False, 'use_exponential_sigmas': False, 'use_beta_sigmas': False, 'steps_offset': 0, 'timestep_spacing': 'linspace' }, 'KDPM2 a': { 'use_karras_sigmas': False, 'use_exponential_sigmas': False, 'use_beta_sigmas': False, 'steps_offset': 0, 'timestep_spacing': 'linspace' }, - 'CMSI': { }, #{ 'sigma_min': 0.002, 'sigma_max': 80.0, 'sigma_data': 0.5, 's_noise': 1.0, 'rho': 7.0, 'clip_denoised': True }, + 'CMSI': { }, } samplers_data_diffusers = [ @@ -225,7 +225,13 @@ class DiffusionSampler: if 'beta_end' in self.config and shared.opts.schedulers_beta_end > 0: self.config['beta_end'] = shared.opts.schedulers_beta_end if 'shift' in self.config: - self.config['shift'] = shared.opts.schedulers_shift + if shared.opts.schedulers_shift == 0: + if 'StableDiffusion3' in model.__class__.__name__: + self.config['shift'] = 3 + if 'Flux' in model.__class__.__name__: + self.config['shift'] = 1 + else: + self.config['shift'] = shared.opts.schedulers_shift if 'use_dynamic_shifting' in self.config: if 'Flux' in model.__class__.__name__: self.config['use_dynamic_shifting'] = shared.opts.schedulers_dynamic_shift diff --git a/modules/shared.py b/modules/shared.py index 46d47b16d..0bd893c9e 100644 --- a/modules/shared.py +++ b/modules/shared.py @@ -801,7 +801,7 @@ options_templates.update(options_section(('sampler-params', "Sampler Settings"), 'schedulers_beta_start': OptionInfo(0, "Beta start", gr.Slider, {"minimum": 0, "maximum": 1, "step": 0.00001, "visible": native}), 'schedulers_beta_end': OptionInfo(0, "Beta end", gr.Slider, {"minimum": 0, "maximum": 1, "step": 0.00001, "visible": native}), 'schedulers_timesteps_range': OptionInfo(1000, "Timesteps range", gr.Slider, {"minimum": 250, "maximum": 4000, "step": 1, "visible": native}), - 'schedulers_shift': OptionInfo(1, "Sampler shift", gr.Slider, {"minimum": 0.1, "maximum": 10, "step": 0.1, "visible": native}), + 'schedulers_shift': OptionInfo(0, "Sampler shift", gr.Slider, {"minimum": 0.1, "maximum": 10, "step": 0.1, "visible": native}), 'schedulers_dynamic_shift': OptionInfo(True, "Sampler dynamic shift", gr.Checkbox, {"visible": native}), # managed from ui.py for backend original k-diffusion