From 9b5f87bf210a0b6ef27643194e220962d1d02541 Mon Sep 17 00:00:00 2001 From: Yi Zhang Date: Mon, 5 Aug 2024 22:27:18 +0800 Subject: [PATCH] update1 --- .../ci_build/github/azure-pipelines/bigmodels-ci-pipeline.yml | 4 ++-- 1 file changed, 2 insertions(+), 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 index 27e7dedad4d65..4ee5dc1084bfd 100644 --- 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/ ; \