may work dropout in LyCORIS #859

2026-04-09 06:45:09 +00:00 · 2023-10-09 14:06:58 +09:00
parent 8b247a330b
commit 025368f51c
1 changed files with 4 additions and 1 deletions
--- a/train_network.py
+++ b/train_network.py
@@ -283,7 +283,10 @@ class NetworkTrainer:
        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,