Update modelling_multi.py
Browse files- 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 |
|