diff --git a/aana/core/models/whisper.py b/aana/core/models/whisper.py index 54c1a615..c46bcfd8 100644 --- a/aana/core/models/whisper.py +++ b/aana/core/models/whisper.py @@ -43,6 +43,10 @@ class WhisperParams(BaseModel): "Temperature for sampling. A single value or a sequence indicating fallback temperatures." ), ) + without_timestamps: bool = Field( + default=False, + description="Whether to sample only text tokens in decoding.", + ) word_timestamps: bool = Field( default=False, description="Whether to extract word-level timestamps." ) @@ -114,6 +118,13 @@ class BatchedWhisperParams(BaseModel): "Temperature for sampling. A single value or a sequence indicating fallback temperatures." ), ) + without_timestamps: bool = Field( + default=True, + description="Whether to sample only text tokens in decoding. Set to 'False' to get phrase-level time stamps (eg. for subtitles)", + ) + word_timestamps: bool = Field( + default=False, description="Whether to extract word-level timestamps." + ) @field_validator("temperature") @classmethod