replace type "GenerationConfig" with "Any" in generate
Browse files
model.py
CHANGED
@@ -1852,7 +1852,7 @@ class CustomWhisperGenerationMixin(WhisperGenerationMixin):
|
|
1852 |
def generate(
|
1853 |
self,
|
1854 |
input_features: Optional[torch.Tensor] = None,
|
1855 |
-
generation_config: Optional[
|
1856 |
logits_processor: Optional[LogitsProcessorList] = None,
|
1857 |
stopping_criteria: Optional[StoppingCriteriaList] = None,
|
1858 |
prefix_allowed_tokens_fn: Optional[Callable[[int, torch.Tensor], List[int]]] = None,
|
|
|
1852 |
def generate(
|
1853 |
self,
|
1854 |
input_features: Optional[torch.Tensor] = None,
|
1855 |
+
generation_config: Optional[Any] = None,
|
1856 |
logits_processor: Optional[LogitsProcessorList] = None,
|
1857 |
stopping_criteria: Optional[StoppingCriteriaList] = None,
|
1858 |
prefix_allowed_tokens_fn: Optional[Callable[[int, torch.Tensor], List[int]]] = None,
|