[Core] Add realtime AR-Diffusion tick sessions for LingBot World 2.0 (#5491)

Signed-off-by: zhangbin.327 <zhangbin.327@bytedance.com>
Signed-off-by: Jack47 <scsvip@gmail.com>
Co-authored-by: zhangbin.327 <zhangbin.327@bytedance.com>
Co-authored-by: Hongsheng Liu <liuhongsheng4@huawei.com>
This commit is contained in:
authored and GitHub committed 2026-08-16 09:11:20 +08:00
1 parent 9798bc3a8d
commit 3875f8ded4
53 files changed
+10754 -129

No files matched your search

@@ -444,6 +444,25 @@ class DiffusionWorker:
else:
profiler.stop()
def _run_ar_diffusion_session_lifecycle(self, method: str, session_id: str) -> bool:
"""Delegate an optional AR session lifecycle call to the model runner."""
if not isinstance(session_id, str) or not session_id.strip():
raise ValueError("session_id must be a non-empty string.")
assert self.model_runner is not None, "Model runner not initialized"
lifecycle_method = getattr(self.model_runner, method, None)
if not callable(lifecycle_method):
return False
lifecycle_method(session_id)
return True
def reset_ar_diffusion_session(self, session_id: str) -> bool:
"""Reset runner-owned AR state through the collective RPC boundary."""
return self._run_ar_diffusion_session_lifecycle("reset_session", session_id)
def close_ar_diffusion_session(self, session_id: str) -> bool:
"""Close runner-owned AR state through the collective RPC boundary."""
return self._run_ar_diffusion_session_lifecycle("close_session", session_id)
def execute_model(
self,
req: OmniDiffusionRequest | list[OmniDiffusionRequest],