From 68550b6b9bf3c1936f30a1857a42a0e49265c784 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=98=89=E6=B8=8A?= <wangjiaming.wjm@alibaba-inc.com>
Date: Mon, 24 Apr 2023 15:23:29 +0800
Subject: [PATCH] update

---
 funasr/lm/abs_model.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/funasr/lm/abs_model.py b/funasr/lm/abs_model.py
index 1f3c8a7b1..560879e8f 100644
--- a/funasr/lm/abs_model.py
+++ b/funasr/lm/abs_model.py
@@ -15,7 +15,7 @@ from typeguard import check_argument_types
 
 from funasr.modules.nets_utils import make_pad_mask
 from funasr.torch_utils.device_funcs import force_gatherable
-from funasr.train.abs_espnet_model import AbsESPnetModel
+from funasr.models.base_model import FunASRModel
 
 class AbsLM(torch.nn.Module, BatchScorerInterface, ABC):
     """The abstract LM class
@@ -39,7 +39,7 @@ class AbsLM(torch.nn.Module, BatchScorerInterface, ABC):
         raise NotImplementedError
 
 
-class LanguageModel(AbsESPnetModel):
+class LanguageModel(FunASRModel):
     def __init__(self, lm: AbsLM, vocab_size: int, ignore_id: int = 0):
         assert check_argument_types()
         super().__init__()