From a1b0cd33d50cee3e4612d1e787399e508b453a4a Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期四, 21 十二月 2023 14:20:21 +0800
Subject: [PATCH] rename register tables
---
funasr/models/seaco_paraformer/model.py | 6 +++---
1 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/funasr/models/seaco_paraformer/model.py b/funasr/models/seaco_paraformer/model.py
index 86aa760..d25babe 100644
--- a/funasr/models/seaco_paraformer/model.py
+++ b/funasr/models/seaco_paraformer/model.py
@@ -51,10 +51,10 @@
from funasr.utils.datadir_writer import DatadirWriter
from funasr.models.paraformer.model import Paraformer
-from funasr.utils.register import register_class, registry_tables
+from funasr.register import tables
-@register_class("model_classes", "SeacoParaformer")
+@tables.register("model_classes", "SeacoParaformer")
class SeacoParaformer(Paraformer):
"""
Author: Speech Lab of DAMO Academy, Alibaba Group
@@ -100,7 +100,7 @@
seaco_decoder = kwargs.get("seaco_decoder", None)
if seaco_decoder is not None:
seaco_decoder_conf = kwargs.get("seaco_decoder_conf")
- seaco_decoder_class = registry_tables.decoder_classes.get(seaco_decoder.lower())
+ seaco_decoder_class = tables.decoder_classes.get(seaco_decoder.lower())
self.seaco_decoder = seaco_decoder_class(
vocab_size=self.vocab_size,
encoder_output_size=self.inner_dim,
--
Gitblit v1.9.1