From a5506d76e817e65685f542efe11cbb070c5df143 Mon Sep 17 00:00:00 2001 From: MichaelClifford Date: Fri, 13 Jan 2023 15:16:22 -0500 Subject: [PATCH] update default ray worker image --- src/codeflare_sdk/cluster/config.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/codeflare_sdk/cluster/config.py b/src/codeflare_sdk/cluster/config.py index 2456ea8d5..4d54b4612 100644 --- a/src/codeflare_sdk/cluster/config.py +++ b/src/codeflare_sdk/cluster/config.py @@ -46,5 +46,5 @@ class ClusterConfiguration: template: str = f"{dir}/templates/new-template.yaml" instascale: bool = False envs: dict = field(default_factory=dict) - image: str = "ghcr.io/ibm-ai-foundation/base:ray1.13.0-py38-gpu-pytorch1.12.0cu116-20220826-202124" + image: str = "ghcr.io/foundation-model-stack/base:ray2.1.0-py38-gpu-pytorch1.12.0cu116-20221213-193103" auth: Authentication = Authentication()