diff --git a/configs/sdxl/scheduler/scheduler_config.json b/configs/sdxl/scheduler/scheduler_config.json index 4b974f7ca..a55b71ccb 100644 --- a/configs/sdxl/scheduler/scheduler_config.json +++ b/configs/sdxl/scheduler/scheduler_config.json @@ -2,7 +2,7 @@ "_class_name": "EulerAncestralDiscreteScheduler", "_diffusers_version": "0.35.1", "beta_end": 0.012, - "beta_schedule": "linear", + "beta_schedule": "scaled_linear", "beta_start": 0.00085, "clip_sample": false, "interpolation_type": "linear", diff --git a/modules/sd_models.py b/modules/sd_models.py index ba465188a..36e05ae12 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -576,8 +576,8 @@ def set_overrides(sd_model, checkpoint_info, model_type): and model_type.startswith("Stable Diffusion") and model_type != "Stable Diffusion 3" ): # SDXL and SD 1.5 scheduler_config = sd_model.scheduler.config - scheduler_config['beta_schedule'] = 'linear' - scheduler_config['timestep_spacing'] = 'trailing' + # scheduler_config['beta_schedule'] = 'scaled_linear' + # scheduler_config['timestep_spacing'] = 'trailing' sd_model.scheduler = diffusers.EulerAncestralDiscreteScheduler.from_config(scheduler_config) if 'bigaspv25' in checkpoint_info_name or ('flow' in checkpoint_info_name and 'flower' not in checkpoint_info_name): scheduler_config = sd_model.scheduler.config