schnell can be trained w/o fast schedule if user provides --i_know_wh…

…at_i_am_doing
bghira · Oct 10, 2024 · 2ba13b1 · 2ba13b1
1 parent 6f5b31e
commit 2ba13b1
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/helpers/configuration/cmd_args.py b/helpers/configuration/cmd_args.py
@@ -2179,8 +2179,10 @@ def parse_cmdline_args(input_args=None):
         "schnell" in args.pretrained_model_name_or_path.lower()
         or args.flux_fast_schedule
     ):
-        if not args.flux_fast_schedule:
-            error_log("Schnell requires --flux_fast_schedule.")
+        if not args.flux_fast_schedule and not args.i_know_what_i_am_doing:
+            error_log(
+                "Schnell requires --flux_fast_schedule (or --i_know_what_i_am_doing)."
+            )
             sys.exit(1)
         flux_version = "schnell"
         model_max_seq_length = 256