Skip to content

Commit 7c7703a

Browse files
committed
Make sure args.base_weights_multiplier items are float
Related to 67372a/LoRA_Easy_Training_Scripts#52
1 parent 1645017 commit 7c7703a

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

train_network.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -894,7 +894,7 @@ def train(self, args):
894894
args.base_weights = [args.base_weights]
895895

896896
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]
897+
args.base_weights_multiplier = [float(x) for x in [args.base_weights_multiplier]]
898898

899899
if args.base_weights is not None:
900900
# base_weights が指定されている場合は、指定された重みを読み込みマージする

0 commit comments

Comments
 (0)