Upload 7 files

modeling_indictrans.py

@@ -61,7 +61,7 @@ def shift_tokens_right(input_ids: torch.Tensor, pad_token_id: int, decoder_start
 
 
 def prepare_decoder_input_ids_label(decoder_input_ids, decoder_attention_mask):
-    labels = decoder_input_ids.full_size(decoder_input_ids.size(), -100)
+    labels = torch.full(decoder_input_ids.size(),-100)
     labels[:, :-1] = decoder_input_ids[:, 1:]
 
     labels_mask = labels == 1