Upload 7 files
Browse files- modeling_indictrans.py +0 -1
modeling_indictrans.py
CHANGED
|
@@ -71,7 +71,6 @@ def prepare_decoder_input_ids_label(decoder_input_ids, decoder_attention_mask):
|
|
| 71 |
decoder_input_ids[mask] = 1
|
| 72 |
decoder_attention_mask[mask] = 0
|
| 73 |
|
| 74 |
-
labels = decoder_input_ids[:, 1:]
|
| 75 |
|
| 76 |
return decoder_input_ids, decoder_attention_mask, labels
|
| 77 |
|
|
|
|
| 71 |
decoder_input_ids[mask] = 1
|
| 72 |
decoder_attention_mask[mask] = 0
|
| 73 |
|
|
|
|
| 74 |
|
| 75 |
return decoder_input_ids, decoder_attention_mask, labels
|
| 76 |
|