diff --git a/examples/dreambooth/train_dreambooth_lora_flux2_klein.py b/examples/dreambooth/train_dreambooth_lora_flux2_klein.py index 7eb627e4bd1d..849b6fcfda08 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(