From 4c1f2da5cd763c872483ea2750c0cd2c02a9d816 Mon Sep 17 00:00:00 2001 From: fclearner <614478391@qq.com> Date: Sun, 4 Aug 2024 15:04:35 +0000 Subject: [PATCH] add lora_ckpt_path to recognize --- wenet/bin/recognize.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/wenet/bin/recognize.py b/wenet/bin/recognize.py index 3101d6eb3..e3654b0ac 100644 --- a/wenet/bin/recognize.py +++ b/wenet/bin/recognize.py @@ -182,6 +182,10 @@ def get_args(): type=bool, default=False, help='''Whether to use lora for biasing''') + parser.add_argument("--lora_ckpt_path", + default=None, + type=str, + help="lora checkpoint path.") args = parser.parse_args() print(args) return args