From 0526a075c567d7508205fe6054310f3d132b3227 Mon Sep 17 00:00:00 2001 From: Emmanuel Schmidbauer Date: Wed, 7 Dec 2022 13:23:16 -0500 Subject: [PATCH] run_speech_recognition_seq2seq.py: add cache_dir param to dataset (#20540) --- .../speech-recognition/run_speech_recognition_seq2seq.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/examples/pytorch/speech-recognition/run_speech_recognition_seq2seq.py b/examples/pytorch/speech-recognition/run_speech_recognition_seq2seq.py index c1466538c3..0487dd43f0 100755 --- a/examples/pytorch/speech-recognition/run_speech_recognition_seq2seq.py +++ b/examples/pytorch/speech-recognition/run_speech_recognition_seq2seq.py @@ -327,6 +327,7 @@ def main(): data_args.dataset_name, data_args.dataset_config_name, split=data_args.train_split_name, + cache_dir=model_args.cache_dir, use_auth_token=True if model_args.use_auth_token else None, ) @@ -335,6 +336,7 @@ def main(): data_args.dataset_name, data_args.dataset_config_name, split=data_args.eval_split_name, + cache_dir=model_args.cache_dir, use_auth_token=True if model_args.use_auth_token else None, )