@@ -1259,6 +1259,9 @@ class ZephyrContext:
12591259 min(max_workers, num_shards), computed at first execute(). If None,
12601260 defaults to os.cpu_count() for LocalClient, or 128 for distributed clients.
12611261 resources: Resource config per worker.
1262+ coordinator_resources: Resource config for the coordinator job. The coordinator
1263+ accumulates scatter manifests for all shards in memory; increase ram for
1264+ large pipelines (many shards). Defaults to 5 GB.
12621265 chunk_storage_prefix: Storage prefix for intermediate chunks. If None, defaults
12631266 to MARIN_PREFIX/tmp/zephyr or /tmp/zephyr.
12641267 name: Descriptive name for this context, used in actor group names for debugging.
@@ -1273,6 +1276,7 @@ class ZephyrContext:
12731276 client : Client | None = None
12741277 max_workers : int | None = None
12751278 resources : ResourceConfig = field (default_factory = lambda : ResourceConfig (cpu = 1 , ram = "1g" ))
1279+ coordinator_resources : ResourceConfig = field (default_factory = lambda : ResourceConfig (cpu = 1 , ram = "5g" ))
12761280 chunk_storage_prefix : str | None = None
12771281 name : str = ""
12781282 no_workers_timeout : float | None = None
@@ -1409,7 +1413,7 @@ def execute(
14091413 _run_coordinator_job ,
14101414 args = (config_path , result_path ),
14111415 ),
1412- resources = ResourceConfig ( cpu = 1 , ram = "1g" ) ,
1416+ resources = self . coordinator_resources ,
14131417 )
14141418 )
14151419
0 commit comments