Spaces:
Running
Running
fix: typo
Browse files
seq2seq/run_seq2seq_flax.py
CHANGED
|
@@ -479,7 +479,7 @@ def main():
|
|
| 479 |
# set up targets
|
| 480 |
# Note: labels correspond to our target indices
|
| 481 |
# decoder input ids are the same but shifted to the right with bos at the beginning (and without last token)
|
| 482 |
-
labels = [[eval(indices) for indices in examples['encoding']]
|
| 483 |
labels = np.asarray(labels)
|
| 484 |
|
| 485 |
# We need the labels, in addition to the decoder_input_ids, for the compute_loss function
|
|
|
|
| 479 |
# set up targets
|
| 480 |
# Note: labels correspond to our target indices
|
| 481 |
# decoder input ids are the same but shifted to the right with bos at the beginning (and without last token)
|
| 482 |
+
labels = [[eval(indices) for indices in examples['encoding']]]
|
| 483 |
labels = np.asarray(labels)
|
| 484 |
|
| 485 |
# We need the labels, in addition to the decoder_input_ids, for the compute_loss function
|