diff --git a/cosyvoice/cli/frontend.py b/cosyvoice/cli/frontend.py index ec4f019..69557d8 100644 --- a/cosyvoice/cli/frontend.py +++ b/cosyvoice/cli/frontend.py @@ -78,6 +78,7 @@ class CosyVoiceFrontEnd: return text_token, text_token_len def _extract_speech_token(self, speech): + assert speech.shape[1] / 16000 <= 30, 'do not support extract speech token for audio longer than 30s' feat = whisper.log_mel_spectrogram(speech, n_mels=128) speech_token = self.speech_tokenizer_session.run(None, {self.speech_tokenizer_session.get_inputs()[0].name: