simple modify

Author: tcaimm

examples/dreambooth/train_dreambooth_lora_flux2.py

@@ -1256,7 +1256,7 @@ def main(args):
     if args.lora_layers is not None:
         target_modules = [layer.strip() for layer in args.lora_layers.split(",")]
     else:
-        # target_modules = ["to_k", "to_q", "to_v", "to_out.0"] - just train transformer_blocks
+        # target_modules = ["to_k", "to_q", "to_v", "to_out.0"] # just train transformer_blocks
 
         # train transformer_blocks and single_transformer_blocks
         target_modules = ["to_k", "to_q", "to_v", "to_out.0"] + [

examples/dreambooth/train_dreambooth_lora_flux2_img2img.py

@@ -1206,7 +1206,7 @@ def main(args):
     if args.lora_layers is not None:
         target_modules = [layer.strip() for layer in args.lora_layers.split(",")]
     else:
-        # target_modules = ["to_k", "to_q", "to_v", "to_out.0"] - just train transformer_blocks
+        # target_modules = ["to_k", "to_q", "to_v", "to_out.0"] # just train transformer_blocks
 
         # train transformer_blocks and single_transformer_blocks
         target_modules = ["to_k", "to_q", "to_v", "to_out.0"] + [