diff --git a/examples/dreambooth/train_dreambooth_lora_flux2_klein.py b/examples/dreambooth/train_dreambooth_lora_flux2_klein.py index cf02c73a1422..2bdd8165f506 100644 --- a/examples/dreambooth/train_dreambooth_lora_flux2_klein.py +++ b/examples/dreambooth/train_dreambooth_lora_flux2_klein.py @@ -359,7 +359,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(