Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 01:04:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

add noobai-rf to flow names

Browse Source
This commit is contained in:
Disty0
2025-11-24 18:21:34 +03:00
parent 4b4d4c8d60
commit 3d7cffc4a4
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/sd_models.py
+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'