From f3c105f088c3fed70308f89d51c029bff0bcb72f 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: a15782cb9f3ee64ba1f5fc2a3da20ac6c6ef0aa0
---
 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":