From 3be66ec19abcff0082e9314d5ef2c4f52b5d154d Mon Sep 17 00:00:00 2001
From: 志浩 <neo.dzh@alibaba-inc.com>
Date: 星期三, 17 五月 2023 15:04:53 +0800
Subject: [PATCH] confirm inference and demo files for SV and SD tasks
---
funasr/tasks/lm.py | 4 +---
1 files changed, 1 insertions(+), 3 deletions(-)
diff --git a/funasr/tasks/lm.py b/funasr/tasks/lm.py
index dc8fd3e..d8ac308 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,4 @@
# 3. Initialize
if args.init is not None:
initialize(model, args.init)
-
- assert check_return_type(model)
return model
--
Gitblit v1.9.1