[diffusion] fix: fix lingbot realtime consistency gt pin (#27080)

This commit is contained in:
Mick
2026-06-03 10:29:37 +08:00
committed by GitHub
parent 3d54056391
commit 71a747cf15
3 changed files with 38 additions and 21 deletions
@@ -324,18 +324,6 @@ LINGBOT_WORLD_REALTIME_sampling_params = DiffusionSamplingParams(
num_frames=9,
fps=16,
realtime_num_chunks=4,
realtime_events=[
{
"after_chunk": 0,
"kind": "camera_actions",
"payload": {"mode": "state", "transitions": [{"actions": ["w"]}]},
},
{
"after_chunk": 2,
"kind": "camera_actions",
"payload": {"mode": "state", "transitions": [{"actions": []}]},
},
],
realtime_perf_thresholds={
"p95_chunk_total_ms": 5000.0,
"p95_scheduler_forward_ms": 4500.0,
@@ -348,6 +336,22 @@ LINGBOT_WORLD_REALTIME_sampling_params = DiffusionSamplingParams(
"guidance_scale": 1.0,
"realtime_causal_sink_size": 9,
"realtime_causal_kv_cache_num_frames": 18,
"condition_inputs": {
"camera_actions": [
["w"],
["w"],
["w"],
["w"],
["w"],
["w"],
[],
[],
[],
[],
[],
[],
]
},
},
)
@@ -33,7 +33,7 @@ if TYPE_CHECKING:
logger = init_logger(__name__)
SGL_TEST_FILES_CI_DATA_REVISION = "8dbe6bc44cda46065fde2168a3d10fd792e2b289"
SGL_TEST_FILES_CI_DATA_REVISION = "ddaad3fca6eba761b0c9692972b8f22b7c463a4d"
SGL_TEST_FILES_CONSISTENCY_GT_ROOT = (
"https://raw.githubusercontent.com/"
f"sgl-project/ci-data/{SGL_TEST_FILES_CI_DATA_REVISION}/"
@@ -504,19 +504,32 @@ def test_lingbot_realtime_plastic_beach_params_are_lossless_gt_ready():
assert params.realtime_perf_ignore_initial_chunks == 2
assert params.realtime_perf_thresholds["p95_chunk_total_ms"] == 5000.0
assert params.realtime_perf_thresholds["p95_scheduler_forward_ms"] == 4500.0
assert params.realtime_events[0]["kind"] == "camera_actions"
assert params.realtime_events[0]["payload"]["mode"] == "state"
assert params.realtime_events == []
assert params.extras["condition_inputs"]["camera_actions"] == [
["w"],
["w"],
["w"],
["w"],
["w"],
["w"],
[],
[],
[],
[],
[],
[],
]
def test_lingbot_realtime_case_is_registered_by_default():
from sglang.multimodal_gen.test.server.gpu_cases import (
ONE_GPU_CASES,
_make_lingbot_realtime_plastic_beach_case,
)
from sglang.multimodal_gen.test.server.gpu_cases import ONE_GPU_CASES
case = _make_lingbot_realtime_plastic_beach_case()
case = next(
item
for item in ONE_GPU_CASES
if item.id == "lingbot_world_realtime_plastic_beach"
)
assert case.id == "lingbot_world_realtime_plastic_beach"
assert case.run_consistency_check is True
assert case.run_perf_check is True
assert case.sampling_params.realtime_output_format is None
assert any(item.id == case.id for item in ONE_GPU_CASES)