Merge pull request #807 from bghira/main

follow-up merge to release
bghira · Aug 18, 2024 · da4bc81 · da4bc81
2 parents 7dd8119 + cabfcfa
commit da4bc81
Show file tree

Hide file tree

Showing 6 changed files with 365 additions and 69 deletions.
diff --git a/helpers/arguments.py b/helpers/arguments.py
@@ -172,6 +172,12 @@ def parse_args(input_args=None):
         type=float,
         default=4.0,
     )
+    parser.add_argument(
+        '--flux_attention_masked_training',
+        action="store_true",
+        default=False,
+        help="Use attention masking while training flux.",
+    )
     parser.add_argument(
         "--smoldit",
         action="store_true",