diff --git a/library/flux_train_utils.py b/library/flux_train_utils.py
index f7f06c5c..3bbd1fad 100644
--- a/library/flux_train_utils.py
+++ b/library/flux_train_utils.py
@@ -617,3 +617,10 @@ def add_flux_train_arguments(parser: argparse.ArgumentParser):
         default=3.0,
         help="Discrete flow shift for the Euler Discrete Scheduler, default is 3.0. / Euler Discrete Schedulerの離散フローシフト、デフォルトは3.0。",
     )
+
+    # bypass guidance module for flux
+    parser.add_argument(
+        "--bypass_flux_guidance"
+        , action="store_true"
+        , help="bypass flux guidance module for Flex.1-Alpha Training / Flex.1-Alpha トレーニング用バイパス フラックス ガイダンス モジュール"
+    )
diff --git a/library/train_util.py b/library/train_util.py
index 3180d694..72b5b24d 100644
--- a/library/train_util.py
+++ b/library/train_util.py
@@ -4103,14 +4103,6 @@ def add_dit_training_arguments(parser: argparse.ArgumentParser):
         "この数を増やすと、トレーニング中のVRAM使用量が減りますが、トレーニング速度（s/it）も低下します。",
     )
 
-    # bypass guidance module for flux
-    parser.add_argument(
-        "--bypass_flux_guidance"
-        , action="store_true"
-        , help="bypass flux guidance module for Flex.1-Alpha Training / Flex.1-Alpha トレーニング用バイパス フラックス ガイダンス モジュール"
-    )
-
-
 
 def get_sanitized_config_or_none(args: argparse.Namespace):
     # if `--log_config` is enabled, return args for logging. if not, return None.