...

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

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

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

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

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

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