add lora_ckpt_path to recognize

wenet-e2e · Aug 4, 2024 · 4c1f2da · 4c1f2da
1 parent ea14646
commit 4c1f2da
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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