Fix seq2seq QA example

2022-09-28 15:45:49 -04:00
parent 64998a57fb
commit 0fc68a7e14
1 changed files with 1 additions and 0 deletions
--- a/examples/pytorch/question-answering/run_seq2seq_qa.py
+++ b/examples/pytorch/question-answering/run_seq2seq_qa.py
@@ -346,6 +346,7 @@ def main():
        raw_datasets = load_dataset(
            extension,
            data_files=data_files,
+            field="data",
            cache_dir=model_args.cache_dir,
            use_auth_token=True if model_args.use_auth_token else None,
        )