update1

microsoft · Aug 5, 2024 · 9b5f87b · 9b5f87b
1 parent 5605aa4
commit 9b5f87b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tools/ci_build/github/azure-pipelines/bigmodels-ci-pipeline.yml b/tools/ci_build/github/azure-pipelines/bigmodels-ci-pipeline.yml
@@ -318,7 +318,7 @@ stages:
         DockerBuildArgs: "
         --build-arg BUILD_UID=$( id -u )
         "
-        Repository: onnxruntimeubi8packagestest_torch
+        Repository: onnxruntimeubi8packagestest_torch1
         UpdateDepsTxt: false
 
     - task: DownloadPackage@1
@@ -334,7 +334,7 @@ stages:
         docker run --rm --gpus all -v $(Build.SourcesDirectory):/workspace \
            -v $(Build.BinariesDirectory)/ort-artifact/:/ort-artifact \
            -v $(Agent.TempDirectory)/meta_llama2_7b_hf:/meta-llama2 \
-           onnxruntimeubi8packagestest_torch \
+           onnxruntimeubi8packagestest_torch1 \
             bash -c "
               set -ex; \
               pushd /workspace/onnxruntime/python/tools/transformers/ ; \