From cb10050cb982d666aa2ff674e0da761846a88662 Mon Sep 17 00:00:00 2001
From: hiyouga <467089858@qq.com>
Date: Sun, 7 Jul 2024 13:10:06 +0800
Subject: [PATCH] fix #4705

Former-commit-id: cfd25c6463bcc263c8672d1de365dd81a028b66a
---
 src/llamafactory/webui/runner.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/llamafactory/webui/runner.py b/src/llamafactory/webui/runner.py
index e23f4d15..a807f3c2 100644
--- a/src/llamafactory/webui/runner.py
+++ b/src/llamafactory/webui/runner.py
@@ -185,7 +185,7 @@ class Runner:
             args["additional_target"] = get("train.additional_target") or None
 
             if args["use_llama_pro"]:
-                args["num_layer_trainable"] = get("train.num_layer_trainable")
+                args["freeze_trainable_layers"] = get("train.freeze_trainable_layers")
 
         # rlhf config
         if args["stage"] == "ppo":