sd3: add skip layer guidance

bghira · Nov 7, 2024 · 48cfc09 · 48cfc09
1 parent 39c05a7
commit 48cfc09
Show file tree

Hide file tree

Showing 3 changed files with 209 additions and 104 deletions.
diff --git a/helpers/configuration/cmd_args.py b/helpers/configuration/cmd_args.py
@@ -1350,6 +1350,14 @@ def get_argument_parser():
             " the default mode, provides the most benefit."
         ),
     )
+    parser.add_argument(
+        "--validation_guidance_skip_layers",
+        type=list,
+        default=None,
+        help=(
+            "StabilityAI recommends a value of [7, 8, 9] for Stable Diffusion 3.5 Medium."
+        ),
+    )
     parser.add_argument(
         "--allow_tf32",
         action="store_true",