Replaced default voice with af_heart as af doesn't exist

This commit is contained in:
Fireblade 2025-02-15 12:36:36 -05:00
parent 8c457c3292
commit 4802128943
7 changed files with 11 additions and 11 deletions

View file

@ -54,7 +54,7 @@ class OpenAISpeechRequest(BaseModel):
)
input: str = Field(..., description="The text to generate audio for")
voice: str = Field(
default="af",
default="af_heart",
description="The voice to use for generation. Can be a base voice or a combined voice name.",
)
response_format: Literal["mp3", "opus", "aac", "flac", "wav", "pcm"] = Field(
@ -98,7 +98,7 @@ class CaptionedSpeechRequest(BaseModel):
)
input: str = Field(..., description="The text to generate audio for")
voice: str = Field(
default="af",
default="af_heart",
description="The voice to use for generation. Can be a base voice or a combined voice name.",
)
response_format: Literal["mp3", "opus", "aac", "flac", "wav", "pcm"] = Field(

View file

@ -34,7 +34,7 @@ def measure_first_token(
json={
"model": "kokoro",
"input": text,
"voice": "af",
"voice": "af_heart",
"response_format": "wav",
"stream": False,
},

View file

@ -34,7 +34,7 @@ def measure_first_token_requests(
json={
"model": "kokoro",
"input": text,
"voice": "af",
"voice": "af_heart",
"response_format": "pcm",
"stream": True,
},
@ -123,7 +123,7 @@ def measure_first_token_openai(
# Make streaming request using OpenAI client
with OPENAI_CLIENT.audio.speech.with_streaming_response.create(
model="kokoro",
voice="af",
voice="af_heart",
response_format="pcm",
input=text,
) as response:

View file

@ -56,7 +56,7 @@ def make_tts_request(
json={
"model": "kokoro",
"input": text,
"voice": "af",
"voice": "af_heart",
"response_format": "wav",
"stream": True,
},
@ -77,7 +77,7 @@ def make_tts_request(
json={
"model": "kokoro",
"input": text,
"voice": "af",
"voice": "af_heart",
"response_format": "wav",
"stream": False,
},

View file

@ -246,7 +246,7 @@ def main():
output_dir.mkdir(exist_ok=True, parents=True)
# First generate audio in each format using the API
voice = "af" # Using default voice
voice = "af_heart" # Using default voice
formats = ["wav", "mp3", "opus", "flac", "pcm"]
stats = []

View file

@ -23,7 +23,7 @@ def test_format(
try:
response = client.audio.speech.create(
model="tts-1", voice="af", input=text, response_format=format
model="tts-1", voice="af_heart", input=text, response_format=format
)
print("Got response, saving to file...")
@ -43,7 +43,7 @@ def test_speed(speed: float):
try:
response = client.audio.speech.create(
model="tts-1",
voice="af",
voice="af_heart",
input="The quick brown fox jumped over the lazy dog.",
response_format="wav",
speed=speed,

View file

@ -16,7 +16,7 @@ async def mock_model_manager():
async def mock_voice_manager():
"""Mock voice manager for UI tests"""
manager = AsyncMock()
manager.list_voices = AsyncMock(return_value=["af", "bm_lewis", "af_sarah"])
manager.list_voices = AsyncMock(return_value=["af_heart", "bm_lewis", "af_sarah"])
return manager