This commit is contained in:
vladmandic
2025-11-24 10:21:46 -05:00
+1 -1
View File
@@ -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'