mrq/ai-voice-cloning
31
74
Fork 43
Code Issues 223 Pull Requests 5 Packages Projects Releases Wiki Activity

fixes

Browse Source
This commit is contained in:
mrq 2023-03-14 17:42:42 +00:00
parent 9d2c7fb942
commit fe03ae5839
2 changed files with 18 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/utils.py
View File

@ -49,6 +49,7 @@ WHISPER_BACKENDS = ["openai/whisper", "lightmare/whispercpp"]
VOCODERS = ['univnet', 'bigvgan_base_24khz_100band', 'bigvgan_24khz_100band'] VOCODERS = ['univnet', 'bigvgan_base_24khz_100band', 'bigvgan_24khz_100band']
TTSES = ['tortoise'] TTSES = ['tortoise']
INFERENCING = False
GENERATE_SETTINGS_ARGS = None GENERATE_SETTINGS_ARGS = None
LEARNING_RATE_SCHEMES = {"Multistep": "MultiStepLR", "Cos. Annealing": "CosineAnnealingLR_Restart"} LEARNING_RATE_SCHEMES = {"Multistep": "MultiStepLR", "Cos. Annealing": "CosineAnnealingLR_Restart"}
@ -320,6 +321,7 @@ def generate(**kwargs):
return info return info
INFERENCING = True
for line, cut_text in enumerate(texts): for line, cut_text in enumerate(texts):
if parameters['emotion'] == "Custom": if parameters['emotion'] == "Custom":
if parameters['prompt'] and parameters['prompt'].strip() != "": if parameters['prompt'] and parameters['prompt'].strip() != "":
@ -371,6 +373,7 @@ def generate(**kwargs):
del gen del gen
do_gc() do_gc()
INFERENCING = False
for k in audio_cache: for k in audio_cache:
audio = audio_cache[k]['audio'] audio = audio_cache[k]['audio']
@ -486,7 +489,11 @@ def generate(**kwargs):
) )
def cancel_generate(): def cancel_generate():
if not INFERENCING:
return
import tortoise.api import tortoise.api
tortoise.api.STOP_SIGNAL = True tortoise.api.STOP_SIGNAL = True
def hash_file(path, algo="md5", buffer_size=0): def hash_file(path, algo="md5", buffer_size=0):

12
src/webui.py
View File

@ -155,6 +155,16 @@ def import_generate_settings_proxy( file=None ):
return tuple(res) return tuple(res)
def reset_generate_settings_proxy( file=None ):
global GENERATE_SETTINGS_ARGS
settings = reset_generate_settings( file )
res = []
for k in GENERATE_SETTINGS_ARGS:
res.append(settings[k] if k in settings else None)
return tuple(res)
def compute_latents_proxy(voice, voice_latents_chunks, progress=gr.Progress(track_tqdm=True)): def compute_latents_proxy(voice, voice_latents_chunks, progress=gr.Progress(track_tqdm=True)):
compute_latents( voice=voice, voice_latents_chunks=voice_latents_chunks, progress=progress ) compute_latents( voice=voice, voice_latents_chunks=voice_latents_chunks, progress=progress )
return voice return voice
@ -688,7 +698,7 @@ def setup_gradio():
) )
reset_generation_settings_button.click( reset_generation_settings_button.click(
fn=reset_generation_settings, fn=reset_generation_settings_proxy,
inputs=None, inputs=None,
outputs=generate_settings outputs=generate_settings
) )