From 0271fbe4fd11114c259ec970b1b04938010a16e3 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期二, 16 五月 2023 18:53:33 +0800
Subject: [PATCH] inference
---
funasr/tasks/lm.py | 5 +----
1 files changed, 1 insertions(+), 4 deletions(-)
diff --git a/funasr/tasks/lm.py b/funasr/tasks/lm.py
index dc8fd3e..1e48655 100644
--- a/funasr/tasks/lm.py
+++ b/funasr/tasks/lm.py
@@ -15,7 +15,7 @@
from funasr.datasets.collate_fn import CommonCollateFn
from funasr.datasets.preprocessor import CommonPreprocessor
from funasr.lm.abs_model import AbsLM
-from funasr.lm.espnet_model import LanguageModel
+from funasr.lm.abs_model import LanguageModel
from funasr.lm.seq_rnn_lm import SequentialRNNLM
from funasr.lm.transformer_lm import TransformerLM
from funasr.tasks.abs_task import AbsTask
@@ -206,6 +206,3 @@
# 3. Initialize
if args.init is not None:
initialize(model, args.init)
-
- assert check_return_type(model)
- return model
--
Gitblit v1.9.1