VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-08 22:35:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

may work dropout in LyCORIS #859

Browse Source
This commit is contained in:
Kohya S
2023-10-09 14:06:58 +09:00
parent 8b247a330b
commit 025368f51c
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
train_network.py
View File

@@ -283,7 +283,10 @@ class NetworkTrainer:
if args.dim_from_weights: if args.dim_from_weights:
network, _ = network_module.create_network_from_weights(1, args.network_weights, vae, text_encoder, unet, **net_kwargs) network, _ = network_module.create_network_from_weights(1, args.network_weights, vae, text_encoder, unet, **net_kwargs)
else: 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( network = network_module.create_network(
1.0, 1.0,
args.network_dim, args.network_dim,