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

Set the default shift to 2.5 for sdxl flowmatch

Browse Source
This commit is contained in:
Disty0
2025-10-24 20:55:07 +03:00
parent 7c2bffd44c
commit d18c040435
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/sd_models.py
+1
View File
@@ -584,6 +584,7 @@ def set_overrides(sd_model, checkpoint_info, model_type):
scheduler_config['prediction_type'] = 'flow_prediction'
scheduler_config['beta_schedule'] = 'linear'
scheduler_config['use_flow_sigmas'] = True
scheduler_config["flow_shift"] = 2.5
sd_model.scheduler = diffusers.UniPCMultistepScheduler.from_config(scheduler_config)
shared.log.info(f'Setting override: model="{checkpoint_info.name}" component=scheduler prediction="flow-prediction"')
elif 'vpred' in checkpoint_info_name or 'v-pred' in checkpoint_info_name or 'v_pred' in checkpoint_info_name: