Добавлены ударения (RUAccent) + нормализация

eremin_tts.py

@@ -31,6 +31,7 @@ import numpy as np
 import soundfile as sf
 import torch
 import torchaudio
+from ruaccent import RUAccent
 
 from f5_tts.infer.utils_infer import (
     infer_process,
@@ -142,11 +143,26 @@ def main():
     model, vocoder = load_f5_model()
     ref_audio_p, ref_text_p = prepare_ref()
 
+    # RUAccent: расстановка ударений
+    print("Loading RUAccent...")
+    accentizer = RUAccent()
+    accentizer.load(omograph_model_size='turbo3.1', use_dictionary=True,
+                    tiny_mode=False)
+    print("  RUAccent loaded.")
+
+    def _accent(text):
+        if not text.strip() or '+' in text:
+            return text
+        try:
+            return accentizer.process_all(text)
+        except Exception:
+            return text
+
     raw_sents = [s.strip() for s in SENT_SPLIT_RE.split(args.text)
                  if s and s.strip()]
     if not raw_sents:
         raw_sents = [args.text]
-    sentences = [normalize_numbers(s) for s in raw_sents]
+    sentences = [_accent(normalize_numbers(s)) for s in raw_sents]
 
     print(f"Предложений: {len(sentences)}")
     for i, s in enumerate(sentences, 1):