Skip to content

Commit 1645017

Browse files
committed
Make args.base_weights_multipler a list if it isn't
Related to 67372a/LoRA_Easy_Training_Scripts#52
1 parent a6c5ea7 commit 1645017

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

train_network.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -893,6 +893,9 @@ def train(self, args):
893893
if args.base_weights is not None and not isinstance(args.base_weights, list):
894894
args.base_weights = [args.base_weights]
895895

896+
if args.base_weights_multiplier is not None and not isinstance(args.base_weights_multiplier, list):
897+
args.base_weights_multiplier = [args.base_weights_multiplier]
898+
896899
if args.base_weights is not None:
897900
# base_weights が指定されている場合は、指定された重みを読み込みマージする
898901
for i, weight_path in enumerate(args.base_weights):

0 commit comments

Comments
 (0)