Skip to content

Commit 0ff65b0

Browse files
update default ray worker image (#46)
1 parent bed3e36 commit 0ff65b0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/codeflare_sdk/cluster/config.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,5 +46,5 @@ class ClusterConfiguration:
4646
template: str = f"{dir}/templates/new-template.yaml"
4747
instascale: bool = False
4848
envs: dict = field(default_factory=dict)
49-
image: str = "ghcr.io/ibm-ai-foundation/base:ray1.13.0-py38-gpu-pytorch1.12.0cu116-20220826-202124"
49+
image: str = "ghcr.io/foundation-model-stack/base:ray2.1.0-py38-gpu-pytorch1.12.0cu116-20221213-193103"
5050
auth: Authentication = Authentication()

0 commit comments

Comments
 (0)