fix: should be converted to array

seq2seq/run_seq2seq_flax.py

@@ -490,7 +490,7 @@ def main():
             jnp.array(labels), config.pad_token_id, config.decoder_start_token_id
         )
 
-        model_inputs["decoder_input_ids"] = decoder_input_ids
+        model_inputs["decoder_input_ids"] = np.asarray(decoder_input_ids)
 
         return model_inputs