From aa8fe52376a04530bd75c5e89d00f8fb97b8dcad Mon Sep 17 00:00:00 2001 From: Libin Tang Date: Thu, 25 Jan 2024 03:47:03 +0200 Subject: [PATCH] This patch is to fix the wav2vec-base ci test as we enabled the hpu_graph_for_training. We need to update ci command too. --- tests/test_examples.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tests/test_examples.py b/tests/test_examples.py index 5e1a1af246..f583074bdb 100644 --- a/tests/test_examples.py +++ b/tests/test_examples.py @@ -420,6 +420,9 @@ def _create_command_line( if "bloom" not in model_name: cmd_line.append("--do_eval") + if "wav2vec-base" in model_name and not deepspeed: + cmd_line.append("--use_hpu_graphs_for_training") + if extra_command_line_arguments is not None: cmd_line += extra_command_line_arguments