fix #4705

2024-07-07 13:10:06 +08:00 · 2024-07-07 13:10:06 +08:00 · a15782cb9f
parent 53b1002fb7
commit a15782cb9f
1 changed files with 1 additions and 1 deletions
--- 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":