From f3fb62cd4fb1f7eebfd862971cc75dd41478d8c2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=98=89=E6=B8=8A?= <wangjiaming.wjm@alibaba-inc.com>
Date: Fri, 16 Jun 2023 13:15:48 +0800
Subject: [PATCH] update repo

---
 funasr/models/e2e_vad.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/funasr/models/e2e_vad.py b/funasr/models/e2e_vad.py
index 14d56a85d..7c55b2e18 100644
--- a/funasr/models/e2e_vad.py
+++ b/funasr/models/e2e_vad.py
@@ -5,6 +5,7 @@ import torch
 from torch import nn
 import math
 from funasr.models.encoder.fsmn_encoder import FSMN
+from funasr.models.base_model import FunASRModel
 
 
 class VadStateMachine(Enum):
@@ -211,7 +212,7 @@ class WindowDetector(object):
         return int(self.frame_size_ms)
 
 
-class E2EVadModel(nn.Module):
+class E2EVadModel(FunASRModel):
     """
     Author: Speech Lab of DAMO Academy, Alibaba Group
     Deep-FSMN for Large Vocabulary Continuous Speech Recognition