From ff83116369ba6afe56430cb0b9d977fbe7f4bced Mon Sep 17 00:00:00 2001
From: Atomie CHEN <atomic_cwh@163.com>
Date: 星期一, 25 三月 2024 09:37:26 +0800
Subject: [PATCH] fix: resolve IndexError when using spk model and the audio contains only 1 segment (#1535)

---
 funasr/models/eend/e2e_diar_eend_ola.py |    2 +-
 1 files changed, 1 insertions(+), 1 deletions(-)

diff --git a/funasr/models/eend/e2e_diar_eend_ola.py b/funasr/models/eend/e2e_diar_eend_ola.py
index 28aa223..ee80836 100644
--- a/funasr/models/eend/e2e_diar_eend_ola.py
+++ b/funasr/models/eend/e2e_diar_eend_ola.py
@@ -7,7 +7,7 @@
 import torch.nn as  nn
 import torch.nn.functional as F
 
-from funasr.models.frontend.wav_frontend import WavFrontendMel23
+from funasr.frontends.wav_frontend import WavFrontendMel23
 from funasr.models.eend.encoder import EENDOLATransformerEncoder
 from funasr.models.eend.encoder_decoder_attractor import EncoderDecoderAttractor
 from funasr.models.eend.utils.losses import standard_loss, cal_power_loss, fast_batch_pit_n_speaker_loss

--
Gitblit v1.9.1