rsxdalv / tts-generation-webui

TTS Generation Web UI (Bark, MusicGen + AudioGen, Tortoise, RVC, Vocos, Demucs, SeamlessM4T, MAGNet, StyleTTS2, MMS)
https://rsxdalv.github.io/tts-generation-webui/
MIT License
1.46k stars 160 forks source link

Stop button #224

Closed BornSaint closed 1 week ago

BornSaint commented 6 months ago

Please, put a stop button to cancel the generation, because sometimes we misclick and have to wait too long to finish

rsxdalv commented 6 months ago

I'll check what I can do. The problem I had before was that I don't think I saw a real way for gradio to implement that. Even now, it might be a hard thing to make because the generation code is deep within. For long generations it's a lot more doable.

On Wed, Dec 6, 2023, 2:23 AM 520Pig520 @.***> wrote:

Please, put a stop button to cancel the generation, because sometimes we misclick and have to wait too long to finish

— Reply to this email directly, view it on GitHub https://github.com/rsxdalv/tts-generation-webui/issues/224, or unsubscribe https://github.com/notifications/unsubscribe-auth/ABTRXI2GDG2MKHTZVR2QMIDYH5RBZAVCNFSM6AAAAABAIEFQHSVHI2DSMVQWIX3LMV43ASLTON2WKOZSGAZDMOBVHAYTGOA . You are receiving this because you are subscribed to this thread.Message ID: @.***>

rsxdalv commented 3 months ago

I have added a way to interrupt batches for MusicGen in the new React UI. I plan to expand this functionality to other generation types, although bark is a bit difficult because the code became way too convoluted. image

rsxdalv commented 1 week ago

React UI uses interrupt as much as it can and generally it's a design goal to be able to stop any long generations. However, this project probably won't allow mid-generation interruptions for the time being as those require some custom code for each and every model.