Skip to content

Commit 545b0c9

Browse files
committed
Throw an error if both --flux_schedule_auto_shift and --flux_schedule_shift are enabled.
1 parent 6a7efd8 commit 545b0c9

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

helpers/training/default_settings/safety_check.py

+9
Original file line numberDiff line numberDiff line change
@@ -107,3 +107,12 @@ def safety_check(args, accelerator):
107107
f"{args.model_type} tuning is not compatible with quantisation. Please set --base_model_precision to 'no_change' or train LyCORIS/LoRA."
108108
)
109109
sys.exit(1)
110+
111+
if (
112+
args.flux_schedule_shift is not None and args.flux_schedule_shift > 0
113+
and args.flux_schedule_auto_shift
114+
):
115+
logger.error(
116+
f"--flux_schedule_auto_shift cannot be combined with --flux_schedule_shift. Please set --flux_schedule_shift to 0 if you want to train with --flux_schedule_auto_shift."
117+
)
118+
sys.exit(1)

0 commit comments

Comments
 (0)