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 fc9c9778edc28..a5e96a26dbfbd 100644
--- a/tools/ci_build/github/azure-pipelines/bigmodels-ci-pipeline.yml
+++ b/tools/ci_build/github/azure-pipelines/bigmodels-ci-pipeline.yml
@@ -162,7 +162,7 @@ stages:
 
     - script: |
         docker run --rm --gpus all -v $PWD:/workspace -v $(Build.BinariesDirectory)/Release:/Release nvcr.io/nvidia/pytorch:22.11-py3 \
-          bash -c "
+          bash -c '
             set -ex; \
             python3 --version; \
             python3 -m pip install --upgrade pip; \
@@ -171,24 +171,25 @@ stages:
             python3 -m pip install -r requirements-cuda11.txt; \
             python3 -m pip install --upgrade polygraphy onnx-graphsurgeon --extra-index-url https://pypi.ngc.nvidia.com; \
             echo Generate an image guided by a text prompt; \
-            python3 demo_txt2img.py --seed 1 --deter-infer 'astronaut riding a horse on mars'; \
-            popd; \
-          "
+            python3 demo_txt2img.py --seed 1 --deter-infer "stronaut riding a horse on mars" ; \
+            popd; '
       displayName: 'Run stable diffusion demo'
       workingDirectory: $(Build.SourcesDirectory)
 
     - script: |
         docker run --rm --gpus all -v $PWD:/workspace nvcr.io/nvidia/pytorch:22.11-py3 \
-          bash -c "
+          bash -c '
             set -ex; \
             python3 --version; \
             python3 -m pip install --upgrade pip; \
-            pushd /workspace/onnxruntime/python/tools/transformers/models/stable_diffusion/test; \
+            pushd /workspace/onnxruntime/python/tools/transformers/models/stable_diffusion/; \
+	          image2=$(find $(pwd) -name "astronaut_riding_a_h*.png") ; \
+            pushd test; \
             python3 -m pip install -r requirements.txt; \
-            echo check generate image; \
-            python3 check_image --image1 "astronaut_riding_txt2image-DDIM-50.png" --image2 ; \
-            popd; \
-          "
+            echo check demo_txt2image.py's generate image; \
+            python3 -u check_image.py --image1 astronaut_riding_txt2image-DDIM-50.png --image2 $image2; \
+	          popd; \
+            popd; '
       displayName: 'Check the generated image'
       workingDirectory: $(Build.SourcesDirectory)