Fix default

jhj0517 · Oct 31, 2024 · 21e9b06 · 21e9b06
1 parent 79d567a
commit 21e9b06
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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"
     )