From 4037a2b5b1278736e566aec12e169100275545ea Mon Sep 17 00:00:00 2001 From: Zach Mueller Date: Tue, 16 Jul 2024 18:18:32 -0400 Subject: [PATCH] SpeechEncoderDecoder doesn't support param buffer assignments (#32009) One more model --- .../speech_encoder_decoder/modeling_speech_encoder_decoder.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/transformers/models/speech_encoder_decoder/modeling_speech_encoder_decoder.py b/src/transformers/models/speech_encoder_decoder/modeling_speech_encoder_decoder.py index e5097b7402bd12..a46a1d62af1163 100644 --- a/src/transformers/models/speech_encoder_decoder/modeling_speech_encoder_decoder.py +++ b/src/transformers/models/speech_encoder_decoder/modeling_speech_encoder_decoder.py @@ -181,6 +181,7 @@ class SpeechEncoderDecoderModel(PreTrainedModel): base_model_prefix = "speech_encoder_decoder" main_input_name = "inputs" supports_gradient_checkpointing = True + _supports_param_buffer_assignment = False def __init__( self,