From ea1fb3d230dc7a9746898a7e435e2c1b4ac85215 Mon Sep 17 00:00:00 2001
From: firestar5683 <168790843+firestar5683@users.noreply.github.com>
Date: Sat, 19 Sep 2026 13:51:17 -0700
Subject: [PATCH] Load semantic planner before diffusion model to limit
 initialization peak

---
 .../experiments/ace_chestnut_20260916/host_hook_adapter.py  | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/roadscore/experiments/ace_chestnut_20260916/host_hook_adapter.py b/roadscore/experiments/ace_chestnut_20260916/host_hook_adapter.py
index b47fc0994d..18e50c1770 100644
--- a/roadscore/experiments/ace_chestnut_20260916/host_hook_adapter.py
+++ b/roadscore/experiments/ace_chestnut_20260916/host_hook_adapter.py
@@ -76,13 +76,13 @@ class HostHookAdapter:
         from acestep.handler import AceStepHandler
         from acestep.llm_inference import LLMHandler
         handler, lm = AceStepHandler(), LLMHandler()
+        status, ok = lm.initialize(str(self.base / 'models/ace/checkpoints'), 'acestep-5Hz-lm-1.7B', backend='mlx', device='mps')
+        if not ok:
+            raise RuntimeError(status)
         status, ok = handler.initialize_service(str(self.base / 'models/ace'), config_path='acestep-v15-turbo',
             device='mps', use_mlx_dit=True, offload_to_cpu=True, offload_dit_to_cpu=False)
         if not ok or not handler.use_mlx_dit or handler.mlx_decoder is None:
             raise RuntimeError(f'No safe MLX preparation boundary: {status}')
-        status, ok = lm.initialize(str(self.base / 'models/ace/checkpoints'), 'acestep-5Hz-lm-1.7B', backend='mlx', device='mps')
-        if not ok:
-            raise RuntimeError(status)
         self.handler, self.lm = handler, lm
 
     def __call__(self, request, sources, output):