Upload 7 files
Browse files- modeling_indictrans.py +1 -1
modeling_indictrans.py
CHANGED
|
@@ -74,7 +74,7 @@ def prepare_decoder_input_ids_label(decoder_input_ids, decoder_attention_mask):
|
|
| 74 |
new_decoder_input_ids[mask] = 1
|
| 75 |
new_decoder_attention_mask[mask] = 0
|
| 76 |
|
| 77 |
-
return
|
| 78 |
|
| 79 |
|
| 80 |
|
|
|
|
| 74 |
new_decoder_input_ids[mask] = 1
|
| 75 |
new_decoder_attention_mask[mask] = 0
|
| 76 |
|
| 77 |
+
return new_decoder_input_ids, new_decoder_attention_mask, labels
|
| 78 |
|
| 79 |
|
| 80 |
|