From e24dbdc496debec225414d4d2c760f5775e64f2a Mon Sep 17 00:00:00 2001
From: 天地 <tiandiweizun@gmail.com>
Date: 星期三, 26 三月 2025 13:44:41 +0800
Subject: [PATCH] 感觉应该从文件读取更合适,因为上面判断了文件存在,且可以读取,如果本身是文本的话,下面也会有逻辑进行处理 (#2452)

---
 funasr/models/uniasr/model.py |   16 ++++++++--------
 1 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/funasr/models/uniasr/model.py b/funasr/models/uniasr/model.py
index 002dcdd..bde6377 100644
--- a/funasr/models/uniasr/model.py
+++ b/funasr/models/uniasr/model.py
@@ -521,10 +521,10 @@
             mask_chunk_predictor = self.encoder.overlap_chunk_cls.get_mask_chunk_predictor(
                 None, device=encoder_out.device, batch_size=encoder_out.size(0)
             )
-            mask_shift_chunk = self.encoder.overlap_chunk_cls.get_mask_shift_chunk(
+            mask_shfit_chunk = self.encoder.overlap_chunk_cls.get_mask_shfit_chunk(
                 None, device=encoder_out.device, batch_size=encoder_out.size(0)
             )
-            encoder_out = encoder_out * mask_shift_chunk
+            encoder_out = encoder_out * mask_shfit_chunk
         pre_acoustic_embeds, pre_token_length, pre_alphas, _ = self.predictor(
             encoder_out,
             ys_out_pad,
@@ -622,10 +622,10 @@
             mask_chunk_predictor = self.encoder2.overlap_chunk_cls.get_mask_chunk_predictor(
                 None, device=encoder_out.device, batch_size=encoder_out.size(0)
             )
-            mask_shift_chunk = self.encoder2.overlap_chunk_cls.get_mask_shift_chunk(
+            mask_shfit_chunk = self.encoder2.overlap_chunk_cls.get_mask_shfit_chunk(
                 None, device=encoder_out.device, batch_size=encoder_out.size(0)
             )
-            encoder_out = encoder_out * mask_shift_chunk
+            encoder_out = encoder_out * mask_shfit_chunk
         pre_acoustic_embeds, pre_token_length, pre_alphas, _ = self.predictor2(
             encoder_out,
             ys_out_pad,
@@ -724,10 +724,10 @@
             mask_chunk_predictor = self.encoder.overlap_chunk_cls.get_mask_chunk_predictor(
                 None, device=encoder_out.device, batch_size=encoder_out.size(0)
             )
-            mask_shift_chunk = self.encoder.overlap_chunk_cls.get_mask_shift_chunk(
+            mask_shfit_chunk = self.encoder.overlap_chunk_cls.get_mask_shfit_chunk(
                 None, device=encoder_out.device, batch_size=encoder_out.size(0)
             )
-            encoder_out = encoder_out * mask_shift_chunk
+            encoder_out = encoder_out * mask_shfit_chunk
         pre_acoustic_embeds, pre_token_length, pre_alphas, _ = self.predictor(
             encoder_out,
             ys_out_pad,
@@ -806,10 +806,10 @@
             mask_chunk_predictor = self.encoder2.overlap_chunk_cls.get_mask_chunk_predictor(
                 None, device=encoder_out.device, batch_size=encoder_out.size(0)
             )
-            mask_shift_chunk = self.encoder2.overlap_chunk_cls.get_mask_shift_chunk(
+            mask_shfit_chunk = self.encoder2.overlap_chunk_cls.get_mask_shfit_chunk(
                 None, device=encoder_out.device, batch_size=encoder_out.size(0)
             )
-            encoder_out = encoder_out * mask_shift_chunk
+            encoder_out = encoder_out * mask_shfit_chunk
         pre_acoustic_embeds, pre_token_length, pre_alphas, _ = self.predictor2(
             encoder_out,
             ys_out_pad,

--
Gitblit v1.9.1