fix: backward compatibility for text_encoder_lr

kohya-ss · Sep 20, 2024 · 0535cd2 · 0535cd2
1 parent b844c70
commit 0535cd2
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/train_network.py b/train_network.py
@@ -471,7 +471,11 @@ def train(self, args):
         if support_multiple_lrs:
             text_encoder_lr = args.text_encoder_lr
         else:
-            text_encoder_lr = None if args.text_encoder_lr is None or len(args.text_encoder_lr) == 0 else args.text_encoder_lr[0]
+            # toml backward compatibility
+            if args.text_encoder_lr is None or isinstance(args.text_encoder_lr, float):
+                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]
         try:
             if support_multiple_lrs:
                 results = network.prepare_optimizer_params_with_multiple_te_lrs(text_encoder_lr, args.unet_lr, args.learning_rate)