From 6e69d784e4814c3dbe35e8f70c6cf4b920c8b20b Mon Sep 17 00:00:00 2001
From: 天地 <tiandiweizun@gmail.com>
Date: 星期三, 19 三月 2025 23:10:13 +0800
Subject: [PATCH] 1. bug fix:list(mean)和list(var)，由于mean和var是numpy，导致写入到文件的格式错误，参考上面的话，大概率是list(mean.tolist())，其实外层list没有必要 (#2437)
---
funasr/models/ct_transformer_streaming/encoder.py | 16 ++++++++--------
1 files changed, 8 insertions(+), 8 deletions(-)
diff --git a/funasr/models/ct_transformer_streaming/encoder.py b/funasr/models/ct_transformer_streaming/encoder.py
index 7d09875..a61319a 100644
--- a/funasr/models/ct_transformer_streaming/encoder.py
+++ b/funasr/models/ct_transformer_streaming/encoder.py
@@ -56,7 +56,7 @@
self.stochastic_depth_rate = stochastic_depth_rate
self.dropout_rate = dropout_rate
- def forward(self, x, mask, cache=None, mask_shift_chunk=None, mask_att_chunk_encoder=None):
+ def forward(self, x, mask, cache=None, mask_shfit_chunk=None, mask_att_chunk_encoder=None):
"""Compute encoded features.
Args:
@@ -93,7 +93,7 @@
self.self_attn(
x,
mask,
- mask_shift_chunk=mask_shift_chunk,
+ mask_shfit_chunk=mask_shfit_chunk,
mask_att_chunk_encoder=mask_att_chunk_encoder,
),
),
@@ -109,7 +109,7 @@
self.self_attn(
x,
mask,
- mask_shift_chunk=mask_shift_chunk,
+ mask_shfit_chunk=mask_shfit_chunk,
mask_att_chunk_encoder=mask_att_chunk_encoder,
)
)
@@ -118,7 +118,7 @@
self.self_attn(
x,
mask,
- mask_shift_chunk=mask_shift_chunk,
+ mask_shfit_chunk=mask_shfit_chunk,
mask_att_chunk_encoder=mask_att_chunk_encoder,
)
)
@@ -132,7 +132,7 @@
if not self.normalize_before:
x = self.norm2(x)
- return x, mask, cache, mask_shift_chunk, mask_att_chunk_encoder
+ return x, mask, cache, mask_shfit_chunk, mask_att_chunk_encoder
def forward_chunk(self, x, cache=None, chunk_size=None, look_back=0):
"""Compute encoded features.
@@ -198,7 +198,7 @@
interctc_layer_idx: List[int] = [],
interctc_use_conditioning: bool = False,
kernel_size: int = 11,
- sanm_shift: int = 0,
+ sanm_shfit: int = 0,
selfattention_layer_type: str = "sanm",
):
super().__init__()
@@ -277,7 +277,7 @@
output_size,
attention_dropout_rate,
kernel_size,
- sanm_shift,
+ sanm_shfit,
)
encoder_selfattn_layer_args = (
@@ -286,7 +286,7 @@
output_size,
attention_dropout_rate,
kernel_size,
- sanm_shift,
+ sanm_shfit,
)
self.encoders0 = repeat(
--
Gitblit v1.9.1