fix: respect frames_after_eos parameter in generate_audio_stream() (#58)

The `frames_after_eos` parameter in `generate_audio_stream()` and
`generate_audio()` was accepted but never used - the code always used
the auto-calculated guess.

**Before:** User passes `frames_after_eos=0` → ignored, uses guess (3+
frames of silence)
**After:** User passes `frames_after_eos=0` → respected, minimal
trailing silence

Default behavior preserved when `frames_after_eos=None`.

I *think* this is the intention of that argument.
This commit is contained in:
BatteryShark
2026-01-21 11:31:25 +01:00
committed by GitHub
parent 2d62797f28
commit 36f21a0a7f
+4 -1
View File
@@ -463,10 +463,13 @@ class TTSModel(nn.Module):
for chunk in chunks: for chunk in chunks:
text_to_generate, frames_after_eos_guess = prepare_text_prompt(chunk) text_to_generate, frames_after_eos_guess = prepare_text_prompt(chunk)
frames_after_eos_guess += 2 frames_after_eos_guess += 2
effective_frames = (
frames_after_eos if frames_after_eos is not None else frames_after_eos_guess
)
yield from self._generate_audio_stream_short_text( yield from self._generate_audio_stream_short_text(
model_state=model_state, model_state=model_state,
text_to_generate=chunk, text_to_generate=chunk,
frames_after_eos=frames_after_eos_guess, frames_after_eos=effective_frames,
copy_state=copy_state, copy_state=copy_state,
) )