ankan8145 commited on
Commit
ee427ec
·
verified ·
1 Parent(s): 77dc7a7

Update modelling_multi.py

Browse files
Files changed (1) hide show
  1. modelling_multi.py +1 -0
modelling_multi.py CHANGED
@@ -245,6 +245,7 @@ class MultiCXREncoderDecoderModel(VisionEncoderDecoderModel):
245
 
246
  if not past_key_values:
247
  token_type_ids = self.token_ids_to_token_type_ids(input_ids, special_token_ids)
 
248
  else:
249
  token_type_ids = self.token_ids_to_token_type_ids_past(input_ids, special_token_ids)
250
 
 
245
 
246
  if not past_key_values:
247
  token_type_ids = self.token_ids_to_token_type_ids(input_ids, special_token_ids)
248
+ decoder_inputs['past_key_values']=None # Because First time there will not be any past_key_values
249
  else:
250
  token_type_ids = self.token_ids_to_token_type_ids_past(input_ids, special_token_ids)
251