mrprimenotes commited on
Commit
106d647
·
verified ·
1 Parent(s): 872634a

fix generate

Browse files
Files changed (1) hide show
  1. model.py +0 -3
model.py CHANGED
@@ -1805,9 +1805,6 @@ class WhisperModel(WhisperPreTrainedModel):
1805
 
1806
  # CUSTOM (patch the generation method)
1807
  class CustomWhisperGenerationMixin(WhisperGenerationMixin):
1808
- def __init__(self, *args, **kwargs):
1809
- super().__init__(*args, **kwargs)
1810
-
1811
  def generate(
1812
  self,
1813
  input_features: Optional[torch.Tensor] = None,
 
1805
 
1806
  # CUSTOM (patch the generation method)
1807
  class CustomWhisperGenerationMixin(WhisperGenerationMixin):
 
 
 
1808
  def generate(
1809
  self,
1810
  input_features: Optional[torch.Tensor] = None,