mirror of
https://github.com/SillyTavern/SillyTavern-Extras.git
synced 2026-04-28 02:11:22 +00:00
cleanup
This commit is contained in:
15
tts_coqui.py
15
tts_coqui.py
@@ -25,6 +25,7 @@ loadedModel = "None"
|
||||
spkdirectory = ""
|
||||
multspeakjson = ""
|
||||
_gpu = False
|
||||
is_coqui_available = os.environ.get("COQUI_STUDIO_TOKEN")
|
||||
|
||||
def setGPU(flag):
|
||||
global _gpu
|
||||
@@ -95,23 +96,9 @@ def load_model(_model, _gpu, _progress):
|
||||
|
||||
if model_type(_config_path) == "tortoise":
|
||||
print("Loading Tortoise...")
|
||||
print("_model", _model)
|
||||
#print("Tortoise not supported at the moment im tired of working on this")
|
||||
|
||||
_loadtortoisemodel = _model_directory.replace("--", "/")
|
||||
print("_loadtortoisemodel", _loadtortoisemodel)
|
||||
|
||||
#config = TortoiseConfig()
|
||||
#model = Tortoise.init_from_config(config)
|
||||
#model.load_checkpoint(config, checkpoint_dir="C:/Users/jsviv/AppData/Local/tts/tts_models--en--multi-dataset--tortoise-v2", eval=False)
|
||||
|
||||
#tts = TTS(_loadtortoisemodel)
|
||||
#tts = TTS(model_name="tts_models/en/multi-dataset/tortoise-v2", progress_bar=True, gpu=True)
|
||||
tts = TTS(_loadtortoisemodel, gpu=_gpu)
|
||||
|
||||
|
||||
loadedModel = _model
|
||||
print("loaded model", loadedModel)
|
||||
|
||||
if model_type(_config_path) == "bark":
|
||||
print("Loading Bark...")
|
||||
|
||||
Reference in New Issue
Block a user