Skip to content

Fix sticky audio prompt #31

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
rkfg wants to merge 1 commit into
base: main
Choose a base branch
from
Open

Fix sticky audio prompt #31

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions chatterbox_node.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -48,6 +48,7 @@ class FL_ChatterboxTTSNode(AudioNodeBase):
"""
_tts_model = None
_tts_device = None
_tts_conds = None

@classmethod
def INPUT_TYPES(cls):
Expand Down Expand Up @@ -146,6 +147,7 @@ def generate_speech(self, text, exaggeration, cfg_weight, temperature, seed, aud
# Load the TTS model or reuse if loaded and device matches
if FL_ChatterboxTTSNode._tts_model is not None and FL_ChatterboxTTSNode._tts_device == device:
tts_model = FL_ChatterboxTTSNode._tts_model
tts_model.conds = FL_ChatterboxTTSNode._tts_conds
message += f"\nReusing loaded TTS model on {device}..."
else:
if FL_ChatterboxTTSNode._tts_model is not None:
Expand All @@ -161,6 +163,7 @@ def generate_speech(self, text, exaggeration, cfg_weight, temperature, seed, aud
message += f"\nLoading TTS model on {device}..."
pbar.update_absolute(10) # Indicate model loading started
tts_model = ChatterboxTTS.from_pretrained(device=device)
FL_ChatterboxTTSNode._tts_conds = tts_model.conds
pbar.update_absolute(50) # Indicate model loading finished

if keep_model_loaded:
Expand Down