From 3b5b8057c6ff84ea8f2faeea69bcbe42ad466189 Mon Sep 17 00:00:00 2001
From: Duo <50307526+iProzd@users.noreply.github.com>
Date: Tue, 26 Mar 2024 19:17:42 +0800
Subject: [PATCH] pt: fix typo in multitask finetune (#3607)

fix #3604 , when doing single-task finetuning from multitask pretrained
model and do not define the finetune model branch from command-line or
input file.
---
 deepmd/pt/utils/finetune.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/deepmd/pt/utils/finetune.py b/deepmd/pt/utils/finetune.py
index 3f76454442..2de4214070 100644
--- a/deepmd/pt/utils/finetune.py
+++ b/deepmd/pt/utils/finetune.py
@@ -146,7 +146,9 @@ def change_finetune_model_params(finetune_model, model_config, model_branch=""):
             model_branch_from=model_branch,
         )
         finetune_links["Default"] = (
-            model_branch if finetune_from_multi_task else "Default"
+            model_config["model_branch_chosen"]
+            if finetune_from_multi_task
+            else "Default"
         )
     else:
         assert model_branch == "", (