mrprimenotes commited on
Commit
02f6a8b
·
verified ·
1 Parent(s): 811659f

replace type "GenerationConfig" with "Any" in generate

Browse files
Files changed (1) hide show
  1. model.py +1 -1
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[GenerationConfig] = 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,
 
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,