From 6e9d84e4aa3402cf7ab59fdabbb6c6a47f471375 Mon Sep 17 00:00:00 2001
From: Yabin Li <wucong.lyb@alibaba-inc.com>
Date: 星期四, 15 六月 2023 17:48:07 +0800
Subject: [PATCH] Update readme.md
---
funasr/bin/diar_infer.py | 7 +++++--
1 files changed, 5 insertions(+), 2 deletions(-)
diff --git a/funasr/bin/diar_infer.py b/funasr/bin/diar_infer.py
index f2dcb1e..4460e3d 100755
--- a/funasr/bin/diar_infer.py
+++ b/funasr/bin/diar_infer.py
@@ -235,8 +235,11 @@
new_seq.append(x)
else:
idx_list = np.where(seq < 2 ** vec_dim)[0]
- idx = np.abs(idx_list - i).argmin()
- new_seq.append(seq[idx_list[idx]])
+ if len(idx_list) > 0:
+ idx = np.abs(idx_list - i).argmin()
+ new_seq.append(seq[idx_list[idx]])
+ else:
+ new_seq.append(0)
return np.row_stack([int2vec(x, vec_dim) for x in new_seq])
def post_processing(self, raw_logits: torch.Tensor, spk_num: int, output_format: str = "speaker_turn"):
--
Gitblit v1.9.1