Skip to content

Commit ac394df

Browse files
authored
Fixed head node configuration to match template (#169)
* Fixed head node configuration to match template * Updated unit tests
1 parent 3b41a22 commit ac394df

File tree

8 files changed

+102
-477
lines changed

8 files changed

+102
-477
lines changed

src/codeflare_sdk/cluster/config.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@ class ClusterConfiguration:
4444
min_memory: int = 2
4545
max_memory: int = 2
4646
gpu: int = 0
47-
template: str = f"{dir}/templates/new-template.yaml"
47+
template: str = f"{dir}/templates/base-template.yaml"
4848
instascale: bool = False
4949
envs: dict = field(default_factory=dict)
5050
image: str = "ghcr.io/foundation-model-stack/base:ray2.1.0-py38-gpu-pytorch1.12.0cu116-20221213-193103"

src/codeflare_sdk/templates/aw-kuberay.yaml

-220
This file was deleted.

0 commit comments

Comments
 (0)