From cf1bf6f5e68ec2794ace0943d6d610067e6625b4 Mon Sep 17 00:00:00 2001 From: Anai-Guo Date: Sun, 12 Apr 2026 23:56:18 -0700 Subject: [PATCH] fix: align text_encoder_out_layers default with pipeline The CLI default [10, 20, 30] does not match the layers actually used in pipeline_flux2_klein.py (9, 18, 27). This off-by-one causes silent incorrect behavior during training. Fixes #13445 --- examples/dreambooth/train_dreambooth_lora_flux2_klein.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/dreambooth/train_dreambooth_lora_flux2_klein.py b/examples/dreambooth/train_dreambooth_lora_flux2_klein.py index 2aa5a1c3e30c..0e8fe6206854 100644 --- a/examples/dreambooth/train_dreambooth_lora_flux2_klein.py +++ b/examples/dreambooth/train_dreambooth_lora_flux2_klein.py @@ -358,7 +358,7 @@ def parse_args(input_args=None): "--text_encoder_out_layers", type=int, nargs="+", - default=[10, 20, 30], + default=[9, 18, 27], help="Text encoder hidden layers to compute the final text embeddings.", ) parser.add_argument(