소스 검색

add prompt contraint

lyuxiang.lx 1 년 전
부모
커밋
bcda6d807c
1개의 변경된 파일1개의 추가작업 그리고 0개의 파일을 삭제
  1. 1 0
      cosyvoice/cli/frontend.py

+ 1 - 0
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: