public-mirrors/Kokoro-FastAPI
1
0
Fork 0
mirror of https://github.com/remsky/Kokoro-FastAPI.git synced 2025-08-21 05:44:06 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add a volume multiplier setting

Browse source 
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
Show stats Download patch file Download diff file Expand all files Collapse all files

1
api/src/core/config.py
View file

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

2
api/src/inference/model_manager.py
View file

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