Skip to content

GitLab

Unverified Commit 788730c6 authored by Vedat Baday's avatar Vedat Baday Committed by GitHub
Browse files

fix(whisper): mutable generation config (#27833)

parent ac975074
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
src/transformers/models/whisper/modeling_whisper.py
View file @ 788730c6
...@@ -14,6 +14,7 @@ ...@@ -14,6 +14,7 @@
# limitations under the License. # limitations under the License.
""" PyTorch Whisper model.""" """ PyTorch Whisper model."""
import copy
import math import math
import warnings import warnings
from typing import Optional, Tuple, Union from typing import Optional, Tuple, Union
...@@ -1933,7 +1934,7 @@ class WhisperForConditionalGeneration(WhisperPreTrainedModel): ...@@ -1933,7 +1934,7 @@ class WhisperForConditionalGeneration(WhisperPreTrainedModel):
) )
if generation_config is None: if generation_config is None:
generation_config = self.generation_config generation_config = copy.deepcopy(self.generation_config)
input_stride = self.model.encoder.conv1.stride[0] * self.model.encoder.conv2.stride[0] input_stride = self.model.encoder.conv1.stride[0] * self.model.encoder.conv2.stride[0]
if num_segment_frames is None: if num_segment_frames is None:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment