spark-tts
commited on
Commit
·
aaaf9fd
1
Parent(s):
42e86e8
fix prompt audio length bug
Browse files
sparktts/models/audio_tokenizer.py
CHANGED
|
@@ -65,11 +65,11 @@ class BiCodecTokenizer:
|
|
| 65 |
|
| 66 |
if ref_segment_length > wav_length:
|
| 67 |
# Repeat and truncate to handle insufficient length
|
| 68 |
-
wav = np.tile(wav,
|
| 69 |
|
| 70 |
return wav[:ref_segment_length]
|
| 71 |
|
| 72 |
-
def process_audio(self, wav_path: Path) -> Tuple[
|
| 73 |
"""load auido and get reference audio from wav path"""
|
| 74 |
wav = load_audio(
|
| 75 |
wav_path,
|
|
|
|
| 65 |
|
| 66 |
if ref_segment_length > wav_length:
|
| 67 |
# Repeat and truncate to handle insufficient length
|
| 68 |
+
wav = np.tile(wav, ref_segment_length // wav_length + 1)
|
| 69 |
|
| 70 |
return wav[:ref_segment_length]
|
| 71 |
|
| 72 |
+
def process_audio(self, wav_path: Path) -> Tuple[np.ndarray, torch.Tensor]:
|
| 73 |
"""load auido and get reference audio from wav path"""
|
| 74 |
wav = load_audio(
|
| 75 |
wav_path,
|