From 7012ca2efc130103c4acd24e3678c7ae280f8db4 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期三, 13 十二月 2023 20:08:55 +0800
Subject: [PATCH] funasr2 paraformer biciparaformer contextuaparaformer
---
funasr/models/model_class_factory.py | 22 +---------------------
1 files changed, 1 insertions(+), 21 deletions(-)
diff --git a/funasr/cli/model_class_factory.py b/funasr/models/model_class_factory.py
similarity index 93%
rename from funasr/cli/model_class_factory.py
rename to funasr/models/model_class_factory.py
index b329492..819ca21 100644
--- a/funasr/cli/model_class_factory.py
+++ b/funasr/models/model_class_factory.py
@@ -123,27 +123,7 @@
default=None,
optional=True,
)
-# model_choices = ClassChoices(
-# "model",
-# classes=dict(
-# asr=ASRModel,
-# uniasr=UniASR,
-# paraformer=Paraformer,
-# paraformer_online=ParaformerOnline,
-# paraformer_bert=ParaformerBert,
-# bicif_paraformer=BiCifParaformer,
-# contextual_paraformer=ContextualParaformer,
-# neatcontextual_paraformer=NeatContextualParaformer,
-# mfcca=MFCCA,
-# timestamp_prediction=TimestampPredictor,
-# rnnt=TransducerModel,
-# rnnt_unified=UnifiedTransducerModel,
-# bat=BATModel,
-# sa_asr=SAASRModel,
-# ),
-# type_check=None,
-# default="asr",
-# )
+
preencoder_choices = ClassChoices(
name="preencoder",
classes=dict(
--
Gitblit v1.9.1