[diffusion] fix: fix lingbot realtime consistency gt pin (#27080)
This commit is contained in:
@@ -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}/"
|
||||
|
||||
+21
-8
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user