From 3b6d92363ef6f37ca375db9431c165dac17e2d5c Mon Sep 17 00:00:00 2001 From: Roberts Slisans Date: Thu, 21 Sep 2023 12:53:29 +0300 Subject: [PATCH 1/2] add continue as semantic history button --- README.md | 2 +- src/bark/BarkModelManager.py | 4 +++- src/bark/generation_tab_bark.py | 23 +++++++++++++++++++++-- 3 files changed, 25 insertions(+), 4 deletions(-) diff --git a/README.md b/README.md index 449d7d5a..0e4aeda2 100644 --- a/README.md +++ b/README.md @@ -39,7 +39,7 @@ https://rsxdalv.github.io/bark-speaker-directory/ ## Changelog Sep 21: * Switch to github docker image storage, new docker image: - * `docker pull ghcr.io/rsxdalv/tts-generation-webui:latest` + * `docker pull ghcr.io/rsxdalv/tts-generation-webui:main` * Fix server_port option in config https://github.com/rsxdalv/tts-generation-webui/issues/168 , thanks to https://github.com/Dartvauder Sep 9: diff --git a/src/bark/BarkModelManager.py b/src/bark/BarkModelManager.py index d8d22c80..0e08f0f7 100644 --- a/src/bark/BarkModelManager.py +++ b/src/bark/BarkModelManager.py @@ -41,10 +41,12 @@ def reload_models(self, config): ) def unload_models(self): - print("Unloading Bark models") + print("Unloading Bark models...") self.models_loaded = False clean_models() + print("Unloaded Bark models") def unload_model(self, model_key): print(f"Unloading Bark model {model_key}") clean_models(model_key=model_key) + print(f"Unloaded Bark model {model_key}") diff --git a/src/bark/generation_tab_bark.py b/src/bark/generation_tab_bark.py index 128daf01..a978de3b 100644 --- a/src/bark/generation_tab_bark.py +++ b/src/bark/generation_tab_bark.py @@ -674,7 +674,11 @@ def update_max_length(value): zip( *[ create_components( - old_generation_dropdown, history_setting, index, seed_input + old_generation_dropdown, + history_prompt_semantic_dropdown, + history_setting, + index, + seed_input, ) for index in range(MAX_OUTPUTS) ] @@ -813,7 +817,13 @@ def insert_npz_file(npz_filename): ] -def create_components(old_generation_dropdown, history_setting, index, seed_input): +def create_components( + old_generation_dropdown, + history_prompt_semantic_dropdown, + history_setting, + index, + seed_input, +): with gr.Column(visible=index == 0) as col: audio = gr.Audio( type="filepath", label="Generated audio", elem_classes="tts-audio" @@ -851,6 +861,7 @@ def create_components(old_generation_dropdown, history_setting, index, seed_inpu ) send_to_vocos_button = gr.Button("Vocos", size="sm") continue_button = gr.Button("Use as history", size="sm") + continue_semantic_button = gr.Button("Use as semantic history", size="sm") npz = gr.State() # type: ignore seed = gr.State() # type: ignore json_text = gr.State() # type: ignore @@ -884,6 +895,14 @@ def create_components(old_generation_dropdown, history_setting, index, seed_inpu outputs=[old_generation_dropdown, history_setting], ) + continue_semantic_button.click( + fn=insert_npz_file, + inputs=[npz], + outputs=[history_prompt_semantic_dropdown, history_setting], + ) + + # fix the bug where selecting No history does not work with burn in prompt + return ( [ audio, From 02acc2ca24ac92fc1ec1732e903642c3e1e12294 Mon Sep 17 00:00:00 2001 From: Roberts Slisans Date: Thu, 21 Sep 2023 12:54:21 +0300 Subject: [PATCH 2/2] readme --- README.md | 1 + 1 file changed, 1 insertion(+) diff --git a/README.md b/README.md index 0e4aeda2..5a39be58 100644 --- a/README.md +++ b/README.md @@ -38,6 +38,7 @@ https://rsxdalv.github.io/bark-speaker-directory/ ## Changelog Sep 21: +* Bark: Add continue as semantic history button * Switch to github docker image storage, new docker image: * `docker pull ghcr.io/rsxdalv/tts-generation-webui:main` * Fix server_port option in config https://github.com/rsxdalv/tts-generation-webui/issues/168 , thanks to https://github.com/Dartvauder