fix name

DTennant · Apr 30, 2024 · 8b40d23 · 8b40d23
1 parent 3a091cd
commit 8b40d23
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/peft/utils/save_and_load.py b/src/peft/utils/save_and_load.py
@@ -306,7 +306,7 @@ def set_peft_model_state_dict(
             PeftType.OFT: "oft_",
             PeftType.POLY: "poly_",
             PeftType.BOFT: "boft_",
-            PeftType.LN_TUNING: "ln_tuning",
+            PeftType.LN_TUNING: "ln_tuning_",
             PeftType.VERA: "vera_lambda_",
         }[config.peft_type]
         for k, v in state_dict.items():