fix prod iamge and target

Signed-off-by: sirutBuasai <sirutbuasai27@outlook.com>

Author: sirutBuasai

.github/workflows/pr-vllm.yml

@@ -133,7 +133,7 @@ jobs:
             IMAGE_URI=${{ needs.build-vllm-ec2-image.outputs.ci-image }}
           else
             AWS_ACCOUNT_ID=${{ vars.PROD_AWS_ACCOUNT_ID }}
-            IMAGE_URI=${{ vars.PROD_AWS_ACCOUNT_ID }}.dkr.ecr.${{ vars.AWS_REGION }}.amazonaws.com/${{ env.PROD_SAGEMAKER_IMAGE }}
+            IMAGE_URI=${{ vars.PROD_AWS_ACCOUNT_ID }}.dkr.ecr.${{ vars.AWS_REGION }}.amazonaws.com/${{ env.PROD_EC2_IMAGE }}
           fi
 
           echo "Image URI to test: ${IMAGE_URI}"
@@ -402,7 +402,7 @@ jobs:
             --cache-to=type=inline \
             --cache-from=type=registry,ref=${CI_IMAGE_URI} \
             --tag ${CI_IMAGE_URI} \
-            --target vllm-ec2 \
+            --target vllm-rayserve-ec2 \
             -f docker/vllm/Dockerfile .
 
       - name: Container push
@@ -435,7 +435,7 @@ jobs:
             IMAGE_URI=${{ needs.build-vllm-rayserve-image.outputs.ci-image }}
           else
             AWS_ACCOUNT_ID=${{ vars.PROD_AWS_ACCOUNT_ID }}
-            IMAGE_URI=${{ vars.PROD_AWS_ACCOUNT_ID }}.dkr.ecr.${{ vars.AWS_REGION }}.amazonaws.com/${{ env.PROD_SAGEMAKER_IMAGE }}
+            IMAGE_URI=${{ vars.PROD_AWS_ACCOUNT_ID }}.dkr.ecr.${{ vars.AWS_REGION }}.amazonaws.com/${{ env.PROD_RAYSERVE_IMAGE }}
           fi
 
           echo "Image URI to test: ${IMAGE_URI}"
@@ -704,7 +704,7 @@ jobs:
             --cache-to=type=inline \
             --cache-from=type=registry,ref=${CI_IMAGE_URI} \
             --tag ${CI_IMAGE_URI} \
-            --target vllm-ec2 \
+            --target vllm-sagemaker
             -f docker/vllm/Dockerfile .
 
       - name: Container push