add proj_mlp to the mix

sayakpaul · sayakpaul · commit 53bf41d2f1bf · 2025-07-08T14:14:13.000+05:30
diff --git a/examples/dreambooth/train_dreambooth_lora_flux_kontext.py b/examples/dreambooth/train_dreambooth_lora_flux_kontext.py
@@ -1393,6 +1393,7 @@ def main(args):
             "ff.net.2",
             "ff_context.net.0.proj",
             "ff_context.net.2",
+            "proj_mlp",
         ]
 
     # now we will add new LoRA weights the transformer layers

Original file line number	Diff line number	Diff line change
`@@ -1393,6 +1393,7 @@ def main(args):`
`1393`	`1393`	`"ff.net.2",`
`1394`	`1394`	`"ff_context.net.0.proj",`
`1395`	`1395`	`"ff_context.net.2",`
	`1396`	`+ "proj_mlp",`
`1396`	`1397`	`]`
`1397`	`1398`
`1398`	`1399`	`# now we will add new LoRA weights the transformer layers`