Skip to content

Commit 7e7b558

Browse files
authored
FIX: lora+: include lr in optimizer kwargs (#1973)
1 parent 1b16753 commit 7e7b558

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/peft/optimizers/loraplus.py

+1
Original file line numberDiff line numberDiff line change
@@ -82,6 +82,7 @@ def create_loraplus_optimizer(
8282
else:
8383
param_groups["groupA"][name] = param
8484

85+
kwargs["lr"] = lr
8586
loraplus_weight_decay = kwargs.pop("loraplus_weight_decay", 0.0)
8687
loraplus_lr_embedding = kwargs.pop("loraplus_lr_embedding", 1e-6)
8788

0 commit comments

Comments
 (0)