diff --git a/README.md b/README.md index 449d7d5a..5a39be58 100644 --- a/README.md +++ b/README.md @@ -38,8 +38,9 @@ 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: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,