From b8e34a13424ddda3d1c00fd03b27c75896a271c7 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=B8=B8=E9=9B=81?= Date: Tue, 16 May 2023 20:12:44 +0800 Subject: [PATCH] inference --- .../{infer.py => demo.py} | 0 funasr/bin/lm_inference_launch.py | 2 -- funasr/tasks/lm.py | 1 + 3 files changed, 1 insertion(+), 2 deletions(-) rename egs_modelscope/lm/speech_transformer_lm_zh-cn-common-vocab8404-pytorch/{infer.py => demo.py} (100%) diff --git a/egs_modelscope/lm/speech_transformer_lm_zh-cn-common-vocab8404-pytorch/infer.py b/egs_modelscope/lm/speech_transformer_lm_zh-cn-common-vocab8404-pytorch/demo.py similarity index 100% rename from egs_modelscope/lm/speech_transformer_lm_zh-cn-common-vocab8404-pytorch/infer.py rename to egs_modelscope/lm/speech_transformer_lm_zh-cn-common-vocab8404-pytorch/demo.py diff --git a/funasr/bin/lm_inference_launch.py b/funasr/bin/lm_inference_launch.py index 0840e6ecd..3d38f6499 100644 --- a/funasr/bin/lm_inference_launch.py +++ b/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 diff --git a/funasr/tasks/lm.py b/funasr/tasks/lm.py index 1e48655d9..d8ac3081a 100644 --- a/funasr/tasks/lm.py +++ b/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