Skip to content

Commit 2c037dd

Browse files
committed
feat: enforce value constraints on top_p and temperature parameters in TTS models
Signed-off-by: James Ding <jamesding365@gmail.com>
1 parent 6c2620d commit 2c037dd

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

src/fishaudio/types/tts.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -64,8 +64,8 @@ class TTSConfig(BaseModel):
6464
prosody: Optional[Prosody] = None
6565

6666
# Model parameters
67-
top_p: float = 0.7
68-
temperature: float = 0.7
67+
top_p: Annotated[float, Field(ge=0.0, le=1.0)] = 0.7
68+
temperature: Annotated[float, Field(ge=0.0, le=1.0)] = 0.7
6969

7070

7171
class TTSRequest(BaseModel):
@@ -87,8 +87,8 @@ class TTSRequest(BaseModel):
8787
normalize: bool = True
8888
latency: LatencyMode = "balanced"
8989
prosody: Optional[Prosody] = None
90-
top_p: float = 0.7
91-
temperature: float = 0.7
90+
top_p: Annotated[float, Field(ge=0.0, le=1.0)] = 0.7
91+
temperature: Annotated[float, Field(ge=0.0, le=1.0)] = 0.7
9292

9393

9494
# WebSocket event types for streaming TTS

0 commit comments

Comments
 (0)