jumbo update with flux.1 refactor, see changelog for details

This commit is contained in:
Vladimir Mandic
2024-09-01 22:56:15 -04:00
parent 0d9ce663e4
commit b4df9a4de1
12 changed files with 248 additions and 123 deletions
+5 -3
View File
@@ -68,8 +68,8 @@ config = {
'Euler EDM': { },
'DPM++ 2M EDM': { 'solver_order': 2, 'solver_type': 'midpoint', 'final_sigmas_type': 'zero', 'algorithm_type': 'dpmsolver++' },
'CMSI': { }, #{ 'sigma_min': 0.002, 'sigma_max': 80.0, 'sigma_data': 0.5, 's_noise': 1.0, 'rho': 7.0, 'clip_denoised': True },
'Euler FlowMatch': { 'timestep_spacing': "linspace", 'shift': 1, },
'Heun FlowMatch': { 'timestep_spacing': "linspace", 'shift': 1, },
'Euler FlowMatch': { 'timestep_spacing': "linspace", 'shift': 1, 'use_dynamic_shifting': False },
'Heun FlowMatch': { 'timestep_spacing': "linspace", 'shift': 1 },
'IPNDM': { },
}
@@ -157,8 +157,10 @@ class DiffusionSampler:
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 'shift' in self.config and shared.opts.schedulers_shift != 1:
if 'shift' in self.config:
self.config['shift'] = shared.opts.schedulers_shift
if 'use_dynamic_shifting' in self.config:
self.config['use_dynamic_shifting'] = shared.opts.schedulers_dynamic_shift
if 'rescale_betas_zero_snr' in self.config:
self.config['rescale_betas_zero_snr'] = shared.opts.schedulers_rescale_betas
if 'timestep_spacing' in self.config and shared.opts.schedulers_timestep_spacing != 'default' and shared.opts.schedulers_timestep_spacing is not None: