Add a volume multiplier setting

Allow configuring output volume via multiplier applied to np array of
audio chunk.

Defaults to 1.0 which is no-op.

Fixes #110
This commit is contained in:
JCallicoat 2025-05-22 06:49:37 -05:00
parent 6f1450c5b4
commit 75963c4aeb
2 changed files with 3 additions and 0 deletions

View file

@ -31,6 +31,7 @@ class Settings(BaseSettings):
# Audio Settings
sample_rate: int = 24000
volume_multiplier: float = 1.0
# Text Processing Settings
target_min_tokens: int = 175 # Target minimum tokens per chunk
target_max_tokens: int = 250 # Target maximum tokens per chunk

View file

@ -141,6 +141,8 @@ Model files not found! You need to download the Kokoro V1 model:
try:
async for chunk in self._backend.generate(*args, **kwargs):
if settings.volume_multiplier != 1.0:
chunk.audio *= settings.volume_multiplier
yield chunk
except Exception as e:
raise RuntimeError(f"Generation failed: {e}")