From c1e365fea09aafda387cac12fdff43d28c598979 Mon Sep 17 00:00:00 2001
From: BienBoy <92378515+BienBoy@users.noreply.github.com>
Date: 星期六, 01 二月 2025 23:29:34 +0800
Subject: [PATCH] fix: resolve unexpected 'out of memory' issue in multi-GPU setup (#2373)
---
funasr/models/sond/encoder/ci_scorers.py | 12 ++++++------
1 files changed, 6 insertions(+), 6 deletions(-)
diff --git a/funasr/models/sond/encoder/ci_scorers.py b/funasr/models/sond/encoder/ci_scorers.py
index c5f45b9..a60560b 100644
--- a/funasr/models/sond/encoder/ci_scorers.py
+++ b/funasr/models/sond/encoder/ci_scorers.py
@@ -7,9 +7,9 @@
super().__init__()
def forward(
- self,
- xs_pad: torch.Tensor,
- spk_emb: torch.Tensor,
+ self,
+ xs_pad: torch.Tensor,
+ spk_emb: torch.Tensor,
):
# xs_pad: B, T, D
# spk_emb: B, N, D
@@ -22,9 +22,9 @@
super().__init__()
def forward(
- self,
- xs_pad: torch.Tensor,
- spk_emb: torch.Tensor,
+ self,
+ xs_pad: torch.Tensor,
+ spk_emb: torch.Tensor,
):
# xs_pad: B, T, D
# spk_emb: B, N, D
--
Gitblit v1.9.1