ken11o2/tortoise-tts
1
0
Fork 0
forked from mrq/tortoise-tts
Code Pull Requests Activity

history tab doesn't naively reuse the voice dir instead for results, experimental "divide total sound size until it fits under requests max chunk size" doesn't have a +1 to mess things up (need to re-evaluate how I want to calculate sizes of bests fits eventually)

Browse Source
This commit is contained in:
mrq 2023-02-14 16:23:04 +00:00
parent 47f4b5bf81
commit b648186691
2 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tortoise/api.py
View File

@ -351,7 +351,7 @@ class TextToSpeech:
chunk_size = int(chunk_size / 2)
print(f"Using method 1: size of best fit: {chunk_size}")
chunks = torch.chunk(concat, int(concat.shape[-1] / chunk_size) + 1, dim=1)
chunks = torch.chunk(concat, int(concat.shape[-1] / chunk_size), dim=1)
# default new behavior: use the smallest voice sample as a common chunk size
else:
if chunk_size is None:

9
webui.py
View File

@ -65,7 +65,7 @@ def generate(
mic = load_audio(mic_audio, tts.input_sample_rate)
voice_samples, conditioning_latents = [mic], None
elif voice == "random":
voice_samples, conditioning_latents = None, None
voice_samples, conditioning_latents = None, tts.get_random_conditioning_latents()
else:
progress(0, desc="Loading voice...")
voice_samples, conditioning_latents = load_voice(voice)
@ -469,9 +469,8 @@ def reload_tts():
def cancel_generate():
tortoise.api.STOP_SIGNAL = True
def get_voice_list():
voice_dir = get_voice_dir()
return sorted([d for d in os.listdir(voice_dir) if os.path.isdir(os.path.join(voice_dir, d))]) + ["microphone", "random"]
def get_voice_list(dir=get_voice_dir()):
return sorted([d for d in os.listdir(dir) if os.path.isdir(os.path.join(dir, d)) and len(os.listdir(os.path.join(dir, d))) > 0 ]) + ["microphone", "random"]
def update_voices():
return gr.Dropdown.update(choices=get_voice_list())
@ -732,7 +731,7 @@ def setup_gradio():
with gr.Row():
with gr.Column():
history_voices = gr.Dropdown(
get_voice_list(),
get_voice_list("./results/"),
label="Voice",
type="value",
)