From f02aea27378dd57c2ced4b28ff9e58ec3876340a Mon Sep 17 00:00:00 2001 From: Quentin Lhoest <42851186+lhoestq@users.noreply.github.com> Date: Fri, 15 Mar 2024 16:35:02 +0100 Subject: [PATCH] Rename `glue` to `nyu-mll/glue` (#29679) * Update run_glue.py * Update run_glue.py * Update run_glue_no_trainer.py --- examples/pytorch/text-classification/run_glue.py | 2 +- examples/pytorch/text-classification/run_glue_no_trainer.py | 2 +- examples/tensorflow/text-classification/run_glue.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/examples/pytorch/text-classification/run_glue.py b/examples/pytorch/text-classification/run_glue.py index 61f81c407c..4825e4944b 100755 --- a/examples/pytorch/text-classification/run_glue.py +++ b/examples/pytorch/text-classification/run_glue.py @@ -304,7 +304,7 @@ def main(): if data_args.task_name is not None: # Downloading and loading a dataset from the hub. raw_datasets = load_dataset( - "glue", + "nyu-mll/glue", data_args.task_name, cache_dir=model_args.cache_dir, token=model_args.token, diff --git a/examples/pytorch/text-classification/run_glue_no_trainer.py b/examples/pytorch/text-classification/run_glue_no_trainer.py index c0086d299c..9ec88206b8 100644 --- a/examples/pytorch/text-classification/run_glue_no_trainer.py +++ b/examples/pytorch/text-classification/run_glue_no_trainer.py @@ -281,7 +281,7 @@ def main(): # download the dataset. if args.task_name is not None: # Downloading and loading a dataset from the hub. - raw_datasets = load_dataset("glue", args.task_name) + raw_datasets = load_dataset("nyu-mll/glue", args.task_name) else: # Loading the dataset from local csv or json file. data_files = {} diff --git a/examples/tensorflow/text-classification/run_glue.py b/examples/tensorflow/text-classification/run_glue.py index f1d0c835b9..1a3143406d 100644 --- a/examples/tensorflow/text-classification/run_glue.py +++ b/examples/tensorflow/text-classification/run_glue.py @@ -265,7 +265,7 @@ def main(): # Downloading and loading a dataset from the hub. In distributed training, the load_dataset function guarantee # that only one local process can concurrently download the dataset. datasets = load_dataset( - "glue", + "nyu-mll/glue", data_args.task_name, cache_dir=model_args.cache_dir, token=model_args.token,