From 3b825e5fc0123117466d4ebfd1fc27fd29cef4ab Mon Sep 17 00:00:00 2001 From: Vara Bonthu Date: Fri, 5 Apr 2024 00:14:12 -0700 Subject: [PATCH] Updated the docker images --- .../stablediffusion-rayserve/ray-serve-stablediffusion.yaml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ai-ml/jark-stack/terraform/examples/inference/stablediffusion-rayserve/ray-serve-stablediffusion.yaml b/ai-ml/jark-stack/terraform/examples/inference/stablediffusion-rayserve/ray-serve-stablediffusion.yaml index a75f20fa0..7194babee 100644 --- a/ai-ml/jark-stack/terraform/examples/inference/stablediffusion-rayserve/ray-serve-stablediffusion.yaml +++ b/ai-ml/jark-stack/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: