Fix discrete_flow_shift default value

2026-04-15 00:32:25 +00:00 · 2026-02-08 07:46:12 +07:00
parent bc6051fdc2
commit 0ebfcef3a5
1 changed files with 2 additions and 2 deletions
--- a/library/anima_train_utils.py
+++ b/library/anima_train_utils.py
@@ -109,8 +109,8 @@ def add_anima_training_arguments(parser: argparse.ArgumentParser):
    parser.add_argument(
        "--discrete_flow_shift",
        type=float,
-        default=3.0,
-        help="Timestep distribution shift for rectified flow training (default: 3.0)",
+        default=1.0,
+        help="Timestep distribution shift for rectified flow training (default: 1.0)",
    )
    parser.add_argument(
        "--timestep_sample_method",