Updated the docker images

awslabs · Apr 5, 2024 · 3b825e5 · 3b825e5
1 parent f5ae88b
commit 3b825e5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...tack/terraform/examples/inference/stablediffusion-rayserve/ray-serve-stablediffusion.yaml b/...tack/terraform/examples/inference/stablediffusion-rayserve/ray-serve-stablediffusion.yaml
@@ -64,7 +64,7 @@ spec:
             # For faster inference scaling, consider building a custom image with only your workload's essential dependencies.
             # Smaller images lead to faster scaling, especially across multiple nodes.
             # Notice that we are using the same image for both the head and worker nodes. You might hit ModuleNotFoundError if you use a different image for head and worker nodes.
-              image: 464566837657.dkr.ecr.us-west-2.amazonaws.com/ray2.10.0-py310-gpu-stablediffusion:v3
+              image: public.ecr.aws/data-on-eks/ray2.10.0-py310-gpu-stablediffusion:latest
               imagePullPolicy: IfNotPresent # Ensure the image is always pulled when updated
               lifecycle:
                 preStop:
@@ -110,7 +110,7 @@ spec:
             # For faster inference scaling, consider building a custom image with only your workload's essential dependencies.
             # Smaller images lead to faster scaling, especially across multiple nodes.
             # Notice that we are using the same image for both the head and worker nodes. You might hit ModuleNotFoundError if you use a different image for head and worker nodes.
-                image: 464566837657.dkr.ecr.us-west-2.amazonaws.com/ray2.10.0-py310-gpu-stablediffusion:v3
+                image: public.ecr.aws/data-on-eks/ray2.10.0-py310-gpu-stablediffusion:latest
                 imagePullPolicy: IfNotPresent # Ensure the image is always pulled when updated
                 lifecycle:
                   preStop: