From 6fbd52693128faff61707c24551ce5a3dbb8f682 Mon Sep 17 00:00:00 2001 From: Kohya S Date: Wed, 31 May 2023 20:23:19 +0900 Subject: [PATCH] show multiplier for base weights to console --- train_network.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/train_network.py b/train_network.py index 8c383b0b..c94d6afc 100644 --- a/train_network.py +++ b/train_network.py @@ -158,13 +158,13 @@ def train(args): if args.base_weights is not None: # 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: multiplier = 1.0 else: multiplier = args.base_weights_multiplier[i] + print(f"merging module: {weight_path} with multiplier {multiplier}") + module, weights_sd = network_module.create_network_from_weights( multiplier, weight_path, vae, text_encoder, unet, for_inference=True )