This commit is contained in:
Yuta Hayashibe 2022-09-29 20:29:45 +09:00
parent 73bd558522
commit 5abd0db07a

View file

@ -58,7 +58,7 @@ class WhisperStreamingTranscriber:
t, t,
prompt, prompt,
beam_size: Optional[int], beam_size: Optional[int],
patience: float, patience: Optional[float],
best_of: Optional[int], best_of: Optional[int],
) -> DecodingOptions: ) -> DecodingOptions:
return DecodingOptions( return DecodingOptions(
@ -93,7 +93,7 @@ class WhisperStreamingTranscriber:
t=t, t=t,
prompt=ctx.buffer_tokens, prompt=ctx.buffer_tokens,
beam_size=self.config.beam_size, beam_size=self.config.beam_size,
patience=0.0, patience=None,
best_of=None, best_of=None,
) )
results: List[DecodingResult] = self.model.decode(segment, _decode_options1) # type: ignore results: List[DecodingResult] = self.model.decode(segment, _decode_options1) # type: ignore