vibe-voice-custom-voices-api

Running on Zero

devignite commited on 5 days ago

Commit

27c5793

verified ·

1 Parent(s): 27c3443

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -142,7 +142,7 @@ def generate_speech_gradio(
             audio_output_tuple = single_speaker_node.generate_speech(
                 text=processed_text,
-                model='VibeVoice-Large',
                 attention_type='auto',
                 free_memory_after_generate=False,
                 diffusion_steps=int(diffusion_steps),
@@ -158,7 +158,7 @@ def generate_speech_gradio(
             logger.info(f"Detected {speaker_count} speakers. Using VibeVoiceMultipleSpeakersNode.")
             audio_output_tuple = multi_speaker_node.generate_speech(
                 text=text,
-                model='VibeVoice-Large',
                 attention_type='auto',
                 free_memory_after_generate=False,
                 diffusion_steps=int(diffusion_steps),

             audio_output_tuple = single_speaker_node.generate_speech(
                 text=processed_text,
+                model='VibeVoice-Large-Q8',
                 attention_type='auto',
                 free_memory_after_generate=False,
                 diffusion_steps=int(diffusion_steps),
             logger.info(f"Detected {speaker_count} speakers. Using VibeVoiceMultipleSpeakersNode.")
             audio_output_tuple = multi_speaker_node.generate_speech(
                 text=text,
+                model='VibeVoice-Large-Q8',
                 attention_type='auto',
                 free_memory_after_generate=False,
                 diffusion_steps=int(diffusion_steps),