mawei/StarPilot
1
0
Fork 0
mirror of https://github.com/firestar5683/StarPilot.git synced 2026-10-04 13:24:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

Load semantic planner before diffusion model to limit initialization peak

Browse Source
This commit is contained in:
firestar5683
2026-09-19 13:51:17 -07:00
parent 9aac0bec15
commit ea1fb3d230
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
roadscore/experiments/ace_chestnut_20260916/host_hook_adapter.py
+3 -3
View File
@@ -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):