anicolson commited on
Commit
fcbc034
·
verified ·
1 Parent(s): 7c30c53

Update modelling_longitudinal.py

Browse files

Fixed issues with attn_implementation and decoder_inputs['past_key_values'].

Files changed (1) hide show
  1. modelling_longitudinal.py +1 -0
modelling_longitudinal.py CHANGED
@@ -127,6 +127,7 @@ class LongitudinalPromptMultiCXREncoderDecoderModel(VisionEncoderDecoderModel):
127
  encoder = MultiCvtWithProjectionHead(config=config.encoder)
128
 
129
  # Decoder:
 
130
  if decoder is None:
131
  decoder = transformers.BertLMHeadModel(config=config.decoder)
132
 
 
127
  encoder = MultiCvtWithProjectionHead(config=config.encoder)
128
 
129
  # Decoder:
130
+ config.decoder._attn_implementation = 'eager'
131
  if decoder is None:
132
  decoder = transformers.BertLMHeadModel(config=config.decoder)
133