From 0876f7302e4e7007ee3a8a632947a70e9b7d7993 Mon Sep 17 00:00:00 2001 From: unknown Date: Fri, 19 Dec 2025 19:37:52 -0800 Subject: [PATCH] ... --- lib/classes/tts_engines/bark.py | 3 ++- lib/classes/tts_engines/fairseq.py | 3 ++- lib/classes/tts_engines/tacotron.py | 3 ++- lib/classes/tts_engines/vits.py | 3 ++- lib/classes/tts_engines/xtts.py | 3 ++- lib/classes/tts_engines/yourtts.py | 3 ++- 6 files changed, 12 insertions(+), 6 deletions(-) diff --git a/lib/classes/tts_engines/bark.py b/lib/classes/tts_engines/bark.py index 3dab0533..27753fe9 100644 --- a/lib/classes/tts_engines/bark.py +++ b/lib/classes/tts_engines/bark.py @@ -9,6 +9,7 @@ from typing import Any from pathlib import Path from huggingface_hub import hf_hub_download +from lib.classes.tts_registry import TTSRegistry from lib.classes.vram_detector import VRAMDetector from lib.classes.tts_engines.common.utils import cleanup_memory, append_sentence2vtt, loaded_tts_size_gb, load_xtts_builtin_list #, ensure_safe_checkpoint from lib.classes.tts_engines.common.audio_filters import detect_gender, trim_audio, normalize_audio, is_audio_data_valid @@ -19,7 +20,7 @@ from lib import * lock = threading.Lock() -class Bark: +class Bark(TTSRegistry, name='bark'): def __init__(self, session:DictProxy): try: self.session = session diff --git a/lib/classes/tts_engines/fairseq.py b/lib/classes/tts_engines/fairseq.py index 9750ce07..ae666d36 100644 --- a/lib/classes/tts_engines/fairseq.py +++ b/lib/classes/tts_engines/fairseq.py @@ -9,6 +9,7 @@ from typing import Any from pathlib import Path from huggingface_hub import hf_hub_download +from lib.classes.tts_registry import TTSRegistry from lib.classes.vram_detector import VRAMDetector from lib.classes.tts_engines.common.utils import cleanup_memory, append_sentence2vtt, loaded_tts_size_gb, load_xtts_builtin_list #, ensure_safe_checkpoint from lib.classes.tts_engines.common.audio_filters import detect_gender, trim_audio, normalize_audio, is_audio_data_valid @@ -19,7 +20,7 @@ from lib import * lock = threading.Lock() -class Fairseq: +class Fairseq(TTSRegistry, name='fairseq'): def __init__(self, session:DictProxy): try: self.session = session diff --git a/lib/classes/tts_engines/tacotron.py b/lib/classes/tts_engines/tacotron.py index 66e1c52c..18460941 100644 --- a/lib/classes/tts_engines/tacotron.py +++ b/lib/classes/tts_engines/tacotron.py @@ -9,6 +9,7 @@ from typing import Any from pathlib import Path from huggingface_hub import hf_hub_download +from lib.classes.tts_registry import TTSRegistry from lib.classes.vram_detector import VRAMDetector from lib.classes.tts_engines.common.utils import cleanup_memory, append_sentence2vtt, loaded_tts_size_gb, load_xtts_builtin_list #, ensure_safe_checkpoint from lib.classes.tts_engines.common.audio_filters import detect_gender, trim_audio, normalize_audio, is_audio_data_valid @@ -19,7 +20,7 @@ from lib import * lock = threading.Lock() -class Tacotron2: +class Tacotron2(TTSRegistry, name='tacotron'): def __init__(self, session:DictProxy): try: self.session = session diff --git a/lib/classes/tts_engines/vits.py b/lib/classes/tts_engines/vits.py index f983b0c4..e8843a49 100644 --- a/lib/classes/tts_engines/vits.py +++ b/lib/classes/tts_engines/vits.py @@ -9,6 +9,7 @@ from typing import Any from pathlib import Path from huggingface_hub import hf_hub_download +from lib.classes.tts_registry import TTSRegistry from lib.classes.vram_detector import VRAMDetector from lib.classes.tts_engines.common.utils import cleanup_memory, append_sentence2vtt, loaded_tts_size_gb, load_xtts_builtin_list #, ensure_safe_checkpoint from lib.classes.tts_engines.common.audio_filters import detect_gender, trim_audio, normalize_audio, is_audio_data_valid @@ -19,7 +20,7 @@ from lib import * lock = threading.Lock() -class Vits: +class Vits(TTSRegistry, name='vits'): def __init__(self, session:DictProxy): try: self.session = session diff --git a/lib/classes/tts_engines/xtts.py b/lib/classes/tts_engines/xtts.py index 321ba094..d1fee1b7 100644 --- a/lib/classes/tts_engines/xtts.py +++ b/lib/classes/tts_engines/xtts.py @@ -9,6 +9,7 @@ from typing import Any from pathlib import Path from huggingface_hub import hf_hub_download +from lib.classes.tts_registry import TTSRegistry from lib.classes.vram_detector import VRAMDetector from lib.classes.tts_engines.common.utils import cleanup_memory, append_sentence2vtt, loaded_tts_size_gb, load_xtts_builtin_list #, ensure_safe_checkpoint from lib.classes.tts_engines.common.audio_filters import detect_gender, trim_audio, normalize_audio, is_audio_data_valid @@ -19,7 +20,7 @@ from lib import * lock = threading.Lock() -class XTTSv2: +class XTTSv2(TTSRegistry, name='xtts'): def __init__(self, session:DictProxy): try: self.session = session diff --git a/lib/classes/tts_engines/yourtts.py b/lib/classes/tts_engines/yourtts.py index 508d87d9..dfaffa53 100644 --- a/lib/classes/tts_engines/yourtts.py +++ b/lib/classes/tts_engines/yourtts.py @@ -9,6 +9,7 @@ from typing import Any from pathlib import Path from huggingface_hub import hf_hub_download +from lib.classes.tts_registry import TTSRegistry from lib.classes.vram_detector import VRAMDetector from lib.classes.tts_engines.common.utils import cleanup_memory, append_sentence2vtt, loaded_tts_size_gb, load_xtts_builtin_list #, ensure_safe_checkpoint from lib.classes.tts_engines.common.audio_filters import detect_gender, trim_audio, normalize_audio, is_audio_data_valid @@ -19,7 +20,7 @@ from lib import * lock = threading.Lock() -class YourTTS: +class YourTTS(TTSRegistry, name='yourtts'): def __init__(self, session:DictProxy): try: self.session = session