From 55c09aeaa25b4bb88a50e09ba68fa6ff00a6d676 Mon Sep 17 00:00:00 2001
From: shixian.shi <shixian.shi@alibaba-inc.com>
Date: 星期一, 15 一月 2024 20:10:39 +0800
Subject: [PATCH] update readme, fix seaco bug

---
 funasr/models/sanm/encoder.py |    4 ++--
 1 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/funasr/models/sanm/encoder.py b/funasr/models/sanm/encoder.py
index 8e159e2..cb4e21a 100644
--- a/funasr/models/sanm/encoder.py
+++ b/funasr/models/sanm/encoder.py
@@ -30,7 +30,7 @@
 
 from funasr.models.ctc.ctc import CTC
 
-from funasr.utils.register import register_class
+from funasr.register import tables
 
 class EncoderLayerSANM(nn.Module):
     def __init__(
@@ -153,7 +153,7 @@
 
         return x, cache
 
-@register_class("encoder_classes", "SANMEncoder")
+@tables.register("encoder_classes", "SANMEncoder")
 class SANMEncoder(nn.Module):
     """
     Author: Speech Lab of DAMO Academy, Alibaba Group

--
Gitblit v1.9.1