From f97e0eb9eee3f14c410ce905b73d0c83033dc1c9 Mon Sep 17 00:00:00 2001
From: 嘉渊 <wangjiaming.wjm@alibaba-inc.com>
Date: 星期五, 28 四月 2023 15:17:38 +0800
Subject: [PATCH] update

---
 funasr/tasks/diar.py |    2 +-
 1 files changed, 1 insertions(+), 1 deletions(-)

diff --git a/funasr/tasks/diar.py b/funasr/tasks/diar.py
index 0fa8c83..6be75a0 100644
--- a/funasr/tasks/diar.py
+++ b/funasr/tasks/diar.py
@@ -106,7 +106,7 @@
         sond=DiarSondModel,
         eend_ola=DiarEENDOLAModel,
     ),
-    type_check=AbsESPnetModel,
+    type_check=FunASRModel,
     default="sond",
 )
 encoder_choices = ClassChoices(

--
Gitblit v1.9.1