From 7d1efe158eda74dc847c397db906f6cb77ac0f84 Mon Sep 17 00:00:00 2001
From: aky15 <ankeyu.aky@11.17.44.249>
Date: 星期三, 12 四月 2023 16:49:56 +0800
Subject: [PATCH] rnnt reorg

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

diff --git a/funasr/models_transducer/decoder/rnn_decoder.py b/funasr/models/rnnt_decoder/rnn_decoder.py
similarity index 97%
rename from funasr/models_transducer/decoder/rnn_decoder.py
rename to funasr/models/rnnt_decoder/rnn_decoder.py
index 04c3228..c4e7951 100644
--- a/funasr/models_transducer/decoder/rnn_decoder.py
+++ b/funasr/models/rnnt_decoder/rnn_decoder.py
@@ -5,8 +5,8 @@
 import torch
 from typeguard import check_argument_types
 
-from funasr.models_transducer.beam_search_transducer import Hypothesis
-from funasr.models_transducer.decoder.abs_decoder import AbsDecoder
+from funasr.modules.beam_search.beam_search_transducer import Hypothesis
+from funasr.models.rnnt_decoder.abs_decoder import AbsDecoder
 from funasr.models.specaug.specaug import SpecAug
 
 class RNNDecoder(AbsDecoder):

--
Gitblit v1.9.1