Update deepmd/pt/entrypoints/main.py

Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>
Signed-off-by: Chenqqian Zhang <[email protected]>

deepmd/pt/entrypoints/main.py

@@ -315,14 +315,14 @@ def main(args: Optional[Union[List[str], argparse.Namespace]] = None):
         if FLAGS.list_model_branch:
             assert (
                 FLAGS.finetune is not None
-            ), "When using --list-model-branch, please use --finetune"
+            ), "Error: The '--list-model-branch' option requires the '--finetune' argument to specify the model."
             state_dict = torch.load(FLAGS.finetune, map_location=env.DEVICE)
             if "model" in state_dict:
                 state_dict = state_dict["model"]
             model_params = state_dict["_extra_state"]["model_params"]
             finetune_from_multi_task = "model_dict" in model_params
             #  Pretrained model must be multitask mode
-            assert finetune_from_multi_task, "When using --list-model-branch, the pretrained model must be multitask model"
+            assert finetune_from_multi_task, "Error: The '--list-model-branch' option requires a multitask pretrained model. The provided model does not meet this criterion."
             model_branch = list(model_params["model_dict"].keys())
             log.info(f"Available model branches are {model_branch}")
         else: