diff --git a/modules/sd_models.py b/modules/sd_models.py index 8a81aafc7..3e8aef5f8 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -657,7 +657,7 @@ def set_overrides(sd_model, checkpoint_info, model_type): # 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): + if 'bigaspv25' in checkpoint_info_name or 'noobai-rf' in checkpoint_info_name ('flow' in checkpoint_info_name and 'flower' not in checkpoint_info_name): scheduler_config = sd_model.scheduler.config scheduler_config['prediction_type'] = 'flow_prediction' scheduler_config['beta_schedule'] = 'linear'