Fix default

2024-10-31 16:33:44 +09:00
parent 79d567af9c
commit 21e9b064a6
1 changed files with 2 additions and 2 deletions
--- a/modules/whisper/data_classes.py
+++ b/modules/whisper/data_classes.py
@@ -293,9 +293,9 @@ class WhisperParams(BaseParams):
        default=True,
        description="Suppress blank outputs at start of sampling"
    )
-    suppress_tokens: Optional[Union[List, str]] = Field(default=[-1], description="Token IDs to suppress")
+    suppress_tokens: Optional[Union[List[int], str]] = Field(default=[-1], description="Token IDs to suppress")
    max_initial_timestamp: float = Field(
-        default=0.0,
+        default=1.0,
        ge=0.0,
        description="Maximum initial timestamp"
    )