Update modelling_longitudinal.py

modelling_longitudinal.py

@@ -270,7 +270,7 @@ class LongitudinalPromptMultiCXREncoderDecoderModel(VisionEncoderDecoderModel):
         if torch.all(input_ids[:, 0] == 1):
             input_ids = input_ids[:, 1:]
 
-        decoder_inputs = self.decoder.prepare_inputs_for_generation(input_ids)
+        decoder_inputs = self.decoder.prepare_inputs_for_generation(input_ids, past_key_values=past_key_values)
         decoder_attention_mask = (input_ids != mask_token_id).int()
         decoder_position_ids = torch.nn.functional.relu(
             torch.cumsum(decoder_attention_mask, dim=1, dtype=torch.int64) - 1