SpeechEncoderDecoder doesn't support param buffer assignments (huggin…

…gface#32009) One more model
amyeroberts · Jul 19, 2024 · 88c85af · 88c85af
1 parent dcdccc4
commit 88c85af
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,