Spaces:
Running
Running
Update tts.py
Browse files
tts.py
CHANGED
|
@@ -263,28 +263,3 @@ def generate_onnx(
|
|
| 263 |
result = torch.tensor(generated_tokens, dtype=torch.long)
|
| 264 |
audio_mask = (result >= AUDIO_OFFSET) & (result < AUDIO_OFFSET + NUM_AUDIO_TOKENS)
|
| 265 |
return result[audio_mask] - AUDIO_OFFSET
|
| 266 |
-
|
| 267 |
-
# ββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 268 |
-
# Quick test
|
| 269 |
-
# ββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 270 |
-
|
| 271 |
-
if __name__ == "__main__":
|
| 272 |
-
|
| 273 |
-
ONNX_DIR = "onnx_export"
|
| 274 |
-
DEVICE = "cpu"
|
| 275 |
-
|
| 276 |
-
TEXT = "ΰΈ ΰΈ²ΰΈΰΈΰΈ΅ΰΉΰΉΰΈΰΈ°ΰΈ₯ΰΈ²ΰΈ’ΰΈΰΈΰΈΰΈ£ΰΈ°ΰΈΰΈΰΉΰΈΰΉΰΉΰΈΰΉΰΈΰΈΰΈ°ΰΉΰΈ£ΰΈΰΈ΅ΰΉΰΈΰΈ΅ΰΈ‘ΰΈ²ΰΈΰΈΰΈ£ΰΈ±ΰΈ ΰΈΰΉΰΈΰΈ₯ΰΈ²ΰΈ’ΰΉΰΈΰΉΰΉΰΈΰΈΰΈΰΈ²ΰΈ‘ΰΉ"
|
| 277 |
-
|
| 278 |
-
onnx_model = OnnxTTSModel(ONNX_DIR, device=DEVICE)
|
| 279 |
-
codec = CodecV6(device=DEVICE)
|
| 280 |
-
ref = codec.encode(r"F:\RVC\F5-TTS-v2\Sample\KARENKAEDE_TH_C.wav")
|
| 281 |
-
speaker_emb = ref["global_embedding"].to(DEVICE)
|
| 282 |
-
|
| 283 |
-
start_time = time.time()
|
| 284 |
-
audio_tokens = generate_onnx(
|
| 285 |
-
onnx_model, TEXT, speaker_emb,
|
| 286 |
-
300, 0.3, 50, 0.95, 1.1
|
| 287 |
-
)
|
| 288 |
-
codec.tokens_to_wav(audio_tokens, speaker_emb, "tts.wav")
|
| 289 |
-
end_time = time.time()
|
| 290 |
-
print(f"Time Used:{round(end_time - start_time,2)}")
|
|
|
|
| 263 |
result = torch.tensor(generated_tokens, dtype=torch.long)
|
| 264 |
audio_mask = (result >= AUDIO_OFFSET) & (result < AUDIO_OFFSET + NUM_AUDIO_TOKENS)
|
| 265 |
return result[audio_mask] - AUDIO_OFFSET
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|