Skip to content

Commit 624152a

Browse files
committed
fix transformers
Signed-off-by: serena-ruan <[email protected]>
1 parent 2dc4edb commit 624152a

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

tests/transformers/helper.py

+4
Original file line numberDiff line numberDiff line change
@@ -222,6 +222,10 @@ def load_whisper_pipeline():
222222
feature_extractor = transformers.WhisperFeatureExtractor.from_pretrained(architecture)
223223
if Version(transformers.__version__) > Version("4.30.2"):
224224
model.generation_config.alignment_heads = [[2, 2], [3, 0], [3, 2], [3, 3], [3, 4], [3, 5]]
225+
if Version(transformers.__version__) > Version("4.49.0"):
226+
# forced_decoder_ids is not allowed
227+
# ref: https://github.com/huggingface/transformers/blob/6a2627918d84f25422b931507a8fb9146106ca20/src/transformers/generation/utils.py#L1083
228+
model.generation_config.forced_decoder_ids = None
225229
return transformers.pipeline(
226230
task=task, model=model, tokenizer=tokenizer, feature_extractor=feature_extractor
227231
)

0 commit comments

Comments
 (0)