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

show multiplier for base weights to console

Browse Source
This commit is contained in:
Kohya S
2023-05-31 20:23:19 +09:00
parent c437dce056
commit 6fbd526931
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
train_network.py
View File

@@ -158,13 +158,13 @@ def train(args):
if args.base_weights is not None: if args.base_weights is not None:
# base_weights が指定されている場合は、指定された重みを読み込みマージする # base_weights が指定されている場合は、指定された重みを読み込みマージする
for i, weight_path in enumerate(args.base_weights): for i, weight_path in enumerate(args.base_weights):
print(f"merging module: {weight_path}")
if args.base_weights_multiplier is None or len(args.base_weights_multiplier) <= i: if args.base_weights_multiplier is None or len(args.base_weights_multiplier) <= i:
multiplier = 1.0 multiplier = 1.0
else: else:
multiplier = args.base_weights_multiplier[i] multiplier = args.base_weights_multiplier[i]
print(f"merging module: {weight_path} with multiplier {multiplier}")
module, weights_sd = network_module.create_network_from_weights( module, weights_sd = network_module.create_network_from_weights(
multiplier, weight_path, vae, text_encoder, unet, for_inference=True multiplier, weight_path, vae, text_encoder, unet, for_inference=True
) )