opencampus
/

sign-whisper-german

Inference Endpoints

Model card Files Files and versions Community

mrprimenotes commited on Jan 12

Commit

9fc192e

·

verified ·

1 Parent(s): 02f6a8b

"StoppingCriteriaList" -> Any

Files changed (1) hide show

model.py +1 -1

model.py CHANGED Viewed

@@ -1854,7 +1854,7 @@ class CustomWhisperGenerationMixin(WhisperGenerationMixin):
         input_features: Optional[torch.Tensor] = None,
         generation_config: Optional[Any] = None,
         logits_processor: Optional[LogitsProcessorList] = None,
-        stopping_criteria: Optional[StoppingCriteriaList] = None,
         prefix_allowed_tokens_fn: Optional[Callable[[int, torch.Tensor], List[int]]] = None,
         synced_gpus: bool = False,
         return_timestamps: Optional[bool] = None,

         input_features: Optional[torch.Tensor] = None,
         generation_config: Optional[Any] = None,
         logits_processor: Optional[LogitsProcessorList] = None,
+        stopping_criteria: Optional[Any] = None,
         prefix_allowed_tokens_fn: Optional[Callable[[int, torch.Tensor], List[int]]] = None,
         synced_gpus: bool = False,
         return_timestamps: Optional[bool] = None,