Skip to content

GitLab

Unverified Commit a94105f9 authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Fix preprocess_function in run_summarization_flax.py (#14769) 


Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent 7e61d56a
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/flax/summarization/run_summarization_flax.py
View file @ a94105f9
...@@ -533,7 +533,7 @@ def main(): ...@@ -533,7 +533,7 @@ def main():
model_inputs["labels"] = labels["input_ids"] model_inputs["labels"] = labels["input_ids"]
decoder_input_ids = shift_tokens_right_fn( decoder_input_ids = shift_tokens_right_fn(
jnp.array(labels["input_ids"]), config.pad_token_id, config.decoder_start_token_id labels["input_ids"], config.pad_token_id, config.decoder_start_token_id
) )
model_inputs["decoder_input_ids"] = np.asarray(decoder_input_ids) model_inputs["decoder_input_ids"] = np.asarray(decoder_input_ids)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment