Spaces:

jhj0517
/

Whisper-WebUI

Running

jhj0517 commited on Sep 23, 2024

Commit

9d581cb

1 Parent(s): db85f5e

Fix default value

Files changed (1) hide show

modules/whisper/whisper_parameter.py CHANGED Viewed

@@ -273,7 +273,7 @@ class WhisperValues:
     diarization_device: str = "cuda"
     length_penalty: float = 1.0
     repetition_penalty: float = 1.0
-    no_repeat_ngram_size: int = 0.0
     prefix: Optional[str] = None
     suppress_blank: bool = True
     suppress_tokens: Optional[str] = "[-1]"

     diarization_device: str = "cuda"
     length_penalty: float = 1.0
     repetition_penalty: float = 1.0
+    no_repeat_ngram_size: int = 0
     prefix: Optional[str] = None
     suppress_blank: bool = True
     suppress_tokens: Optional[str] = "[-1]"