riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 10:12:23 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Vedat Baday 2023-12-05 21:01:07 +03:00 committed by GitHub
parent ac975074e6
commit 788730c670
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/transformers/models/whisper/modeling_whisper.py
View File

@ -14,6 +14,7 @@
# limitations under the License.
""" PyTorch Whisper model."""
import copy
import math
import warnings
from typing import Optional, Tuple, Union
@ -1933,7 +1934,7 @@ class WhisperForConditionalGeneration(WhisperPreTrainedModel):
)
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]
if num_segment_frames is None: