mirror of https://github.com/vladmandic/automatic
sdxl default to scaled_linear beta
Signed-off-by: Vladimir Mandic <mandic00@live.com>pull/4312/head
parent
e3bae227d5
commit
4ecec822e0
|
|
@ -2,7 +2,7 @@
|
||||||
"_class_name": "EulerAncestralDiscreteScheduler",
|
"_class_name": "EulerAncestralDiscreteScheduler",
|
||||||
"_diffusers_version": "0.35.1",
|
"_diffusers_version": "0.35.1",
|
||||||
"beta_end": 0.012,
|
"beta_end": 0.012,
|
||||||
"beta_schedule": "linear",
|
"beta_schedule": "scaled_linear",
|
||||||
"beta_start": 0.00085,
|
"beta_start": 0.00085,
|
||||||
"clip_sample": false,
|
"clip_sample": false,
|
||||||
"interpolation_type": "linear",
|
"interpolation_type": "linear",
|
||||||
|
|
|
||||||
|
|
@ -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"
|
and model_type.startswith("Stable Diffusion") and model_type != "Stable Diffusion 3"
|
||||||
): # SDXL and SD 1.5
|
): # SDXL and SD 1.5
|
||||||
scheduler_config = sd_model.scheduler.config
|
scheduler_config = sd_model.scheduler.config
|
||||||
scheduler_config['beta_schedule'] = 'linear'
|
# scheduler_config['beta_schedule'] = 'scaled_linear'
|
||||||
scheduler_config['timestep_spacing'] = 'trailing'
|
# scheduler_config['timestep_spacing'] = 'trailing'
|
||||||
sd_model.scheduler = diffusers.EulerAncestralDiscreteScheduler.from_config(scheduler_config)
|
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):
|
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
|
scheduler_config = sd_model.scheduler.config
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue