diff --git a/src/codeflare_sdk/cluster/config.py b/src/codeflare_sdk/cluster/config.py index cd49feb37..921ffb16b 100644 --- a/src/codeflare_sdk/cluster/config.py +++ b/src/codeflare_sdk/cluster/config.py @@ -15,4 +15,4 @@ class ClusterConfiguration: template: str = "src/codeflare_sdk/templates/new-template.yaml" instascale: bool = False envs: dict = {} - image: str = "rayproject/ray:latest" + image: str = "ghcr.io/ibm-ai-foundation/base:ray1.13.0-py38-gpu-pytorch1.12.0cu116-20220826-202124"