瀏覽代碼

inference

游雁 2 年之前
父節點
當前提交
b8e34a1342

+ 0 - 0
egs_modelscope/lm/speech_transformer_lm_zh-cn-common-vocab8404-pytorch/infer.py → egs_modelscope/lm/speech_transformer_lm_zh-cn-common-vocab8404-pytorch/demo.py


+ 0 - 2
funasr/bin/lm_inference_launch.py

@@ -3,8 +3,6 @@
 # Copyright FunASR (https://github.com/alibaba-damo-academy/FunASR). All Rights Reserved.
 #  MIT License  (https://opensource.org/licenses/MIT)
 
-
-
 import argparse
 import logging
 import os

+ 1 - 0
funasr/tasks/lm.py

@@ -206,3 +206,4 @@ class LMTask(AbsTask):
         # 3. Initialize
         if args.init is not None:
             initialize(model, args.init)
+        return model