VIZINTZOR commited on
Commit
e6a1e4e
Β·
1 Parent(s): ddfff69

Update tts.py

Browse files
Files changed (1) hide show
  1. tts.py +0 -25
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