may work dropout in LyCORIS #859

kohya-ss · Oct 9, 2023 · 025368f · 025368f
1 parent 8b247a3
commit 025368f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/train_network.py b/train_network.py
@@ -283,7 +283,10 @@ def train(self, args):
         if args.dim_from_weights:
             network, _ = network_module.create_network_from_weights(1, args.network_weights, vae, text_encoder, unet, **net_kwargs)
         else:
-            # LyCORIS will work with this...
+            if "dropout" not in net_kwargs:
+                # workaround for LyCORIS (;^ω^)
+                net_kwargs["dropout"] = args.network_dropout
+
             network = network_module.create_network(
                 1.0,
                 args.network_dim,