SpeechEncoderDecoder doesn't support param buffer assignments (#32009)

One more model
huggingface · Jul 16, 2024 · 4037a2b · 4037a2b
1 parent 6f40a21
commit 4037a2b
Showing 1 changed file with 1 addition and 0 deletions.
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
@@ -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,