From 5eec24837ffa4d0519c242896f46a02d25c8e412 Mon Sep 17 00:00:00 2001 From: Baiju Meswani Date: Wed, 14 Jun 2023 20:36:16 -0700 Subject: [PATCH] Fix for AMD GPU pipeline (#16357) --- .../github/azure-pipelines/orttraining-pai-ci-pipeline.yml | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tools/ci_build/github/azure-pipelines/orttraining-pai-ci-pipeline.yml b/tools/ci_build/github/azure-pipelines/orttraining-pai-ci-pipeline.yml index f8d2bb3f88..93aae2fb95 100644 --- a/tools/ci_build/github/azure-pipelines/orttraining-pai-ci-pipeline.yml +++ b/tools/ci_build/github/azure-pipelines/orttraining-pai-ci-pipeline.yml @@ -239,8 +239,7 @@ jobs: python -m onnxruntime.training.ortmodule.torch_cpp_extensions.install; \ python orttraining_ortmodule_tests.py \ --mnist /mnist \ - --bert_data /bert_data/hf_data/glue_data/CoLA/original/raw \ - --transformers_cache /hf_models_cache/huggingface/transformers" + --bert_data /bert_data/hf_data/glue_data/CoLA/original/raw" workingDirectory: $(Build.SourcesDirectory) displayName: 'Run orttraining_ortmodule_tests.py' condition: and(succeededOrFailed(), eq(variables.onnxruntimeBuildSucceeded, 'true'))