Fix for AMD GPU pipeline (microsoft#16357)

JiazhouLvguan · Jun 15, 2023 · 5eec248 · 5eec248
1 parent 73dad44
commit 5eec248
Showing 1 changed file with 1 addition and 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
@@ -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'))