fix generate
Browse files
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,
|