From 51c8dc1c596ea23d5fe0565e5f6cd02eb702ad93 Mon Sep 17 00:00:00 2001 From: tastelikefeet Date: Fri, 27 Feb 2026 16:48:12 +0800 Subject: [PATCH] add more comment for node resouces --- src/twinkle/infra/_ray/resource_manager.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/twinkle/infra/_ray/resource_manager.py b/src/twinkle/infra/_ray/resource_manager.py index d6fea8f1..09149503 100644 --- a/src/twinkle/infra/_ray/resource_manager.py +++ b/src/twinkle/infra/_ray/resource_manager.py @@ -99,7 +99,10 @@ def __init__(self, nproc_per_node: int, ncpu_proc_per_node: int, groups: List[De if cpu_proc_count > 0: cpu_nnodes = math.ceil(cpu_proc_count / ncpu_proc_per_node) assert cpu_nnodes <= len(self.nodes), (f'Not enough nodes for CPU processes, required nodes: {cpu_nnodes}, ' - f'available: {len(self.nodes)}') + f'available: {len(self.nodes)}, ensure your ray cluster ' + f'has enough nodes, or ' + f'setting `twinkle.initialize(nproc_per_node=N(default 8), ...) ' + f'to reduce per node GPU usage.`') for i in range(cpu_nnodes): node = self.nodes[i] node_cpu = int(node['Resources']['CPU'])