From 3fcb5dcfede0103d2465dd85a2ef6b77af58c7bc Mon Sep 17 00:00:00 2001
From: shixian.shi <shixian.shi@alibaba-inc.com>
Date: 星期一, 15 一月 2024 20:25:35 +0800
Subject: [PATCH] update scripts
---
funasr/models/sanm/model.py | 4 ++--
1 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/funasr/models/sanm/model.py b/funasr/models/sanm/model.py
index e01394c..d51478f 100644
--- a/funasr/models/sanm/model.py
+++ b/funasr/models/sanm/model.py
@@ -3,9 +3,9 @@
import torch
from funasr.models.transformer.model import Transformer
-from funasr.utils.register import register_class, registry_tables
+from funasr.register import tables
-@register_class("model_classes", "SANM")
+@tables.register("model_classes", "SANM")
class SANM(Transformer):
"""CTC-attention hybrid Encoder-Decoder model"""
--
Gitblit v1.9.1