Skip to content

GitLab

Unverified Commit 4037a2b5 authored by Zach Mueller's avatar Zach Mueller Committed by GitHub
Browse files

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

One more model
parent 6f40a213
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
src/transformers/models/speech_encoder_decoder/modeling_speech_encoder_decoder.py
View file @ 4037a2b5
...@@ -181,6 +181,7 @@ class SpeechEncoderDecoderModel(PreTrainedModel): ...@@ -181,6 +181,7 @@ class SpeechEncoderDecoderModel(PreTrainedModel):
base_model_prefix = "speech_encoder_decoder" base_model_prefix = "speech_encoder_decoder"
main_input_name = "inputs" main_input_name = "inputs"
supports_gradient_checkpointing = True supports_gradient_checkpointing = True
_supports_param_buffer_assignment = False
def __init__( def __init__(
self, self,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment