Skip to content

Commit

Permalink
add compatibility for int LR (D-Adaptation etc.) kohya-ss#1620
Browse files Browse the repository at this point in the history
  • Loading branch information
kohya-ss committed Sep 20, 2024
1 parent 24f8975 commit 583d4a4
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion train_network.py
Original file line number Diff line number Diff line change
Expand Up @@ -472,7 +472,7 @@ def train(self, args):
text_encoder_lr = args.text_encoder_lr
else:
# toml backward compatibility
if args.text_encoder_lr is None or isinstance(args.text_encoder_lr, float):
if args.text_encoder_lr is None or isinstance(args.text_encoder_lr, float) or isinstance(args.text_encoder_lr, int):
text_encoder_lr = args.text_encoder_lr
else:
text_encoder_lr = None if len(args.text_encoder_lr) == 0 else args.text_encoder_lr[0]
Expand Down

0 comments on commit 583d4a4

Please sign in to comment.