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

Official weights to LoRA

Browse Source
This commit is contained in:
Kohya S
2023-02-13 23:38:38 +09:00
parent bc9fc4ccee
commit cebee02698
5 changed files with 559 additions and 130 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
train_control_net.py
View File

@@ -380,7 +380,7 @@ def train(args):
net_kwargs[key] = value
# if a new network is added in future, add if ~ then blocks for each network (;'∀')
network: control_net_rola.LoRANetwork = network_module.create_network(
network: control_net_rola.ControlLoRANetwork = network_module.create_network(
1.0, args.network_dim, args.network_alpha, vae, text_encoder, unet, **net_kwargs)
if network is None:
return