[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,
|
num_frames=9,
|
||||||
fps=16,
|
fps=16,
|
||||||
realtime_num_chunks=4,
|
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={
|
realtime_perf_thresholds={
|
||||||
"p95_chunk_total_ms": 5000.0,
|
"p95_chunk_total_ms": 5000.0,
|
||||||
"p95_scheduler_forward_ms": 4500.0,
|
"p95_scheduler_forward_ms": 4500.0,
|
||||||
@@ -348,6 +336,22 @@ LINGBOT_WORLD_REALTIME_sampling_params = DiffusionSamplingParams(
|
|||||||
"guidance_scale": 1.0,
|
"guidance_scale": 1.0,
|
||||||
"realtime_causal_sink_size": 9,
|
"realtime_causal_sink_size": 9,
|
||||||
"realtime_causal_kv_cache_num_frames": 18,
|
"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__)
|
logger = init_logger(__name__)
|
||||||
|
|
||||||
SGL_TEST_FILES_CI_DATA_REVISION = "8dbe6bc44cda46065fde2168a3d10fd792e2b289"
|
SGL_TEST_FILES_CI_DATA_REVISION = "ddaad3fca6eba761b0c9692972b8f22b7c463a4d"
|
||||||
SGL_TEST_FILES_CONSISTENCY_GT_ROOT = (
|
SGL_TEST_FILES_CONSISTENCY_GT_ROOT = (
|
||||||
"https://raw.githubusercontent.com/"
|
"https://raw.githubusercontent.com/"
|
||||||
f"sgl-project/ci-data/{SGL_TEST_FILES_CI_DATA_REVISION}/"
|
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_ignore_initial_chunks == 2
|
||||||
assert params.realtime_perf_thresholds["p95_chunk_total_ms"] == 5000.0
|
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_perf_thresholds["p95_scheduler_forward_ms"] == 4500.0
|
||||||
assert params.realtime_events[0]["kind"] == "camera_actions"
|
assert params.realtime_events == []
|
||||||
assert params.realtime_events[0]["payload"]["mode"] == "state"
|
assert params.extras["condition_inputs"]["camera_actions"] == [
|
||||||
|
["w"],
|
||||||
|
["w"],
|
||||||
|
["w"],
|
||||||
|
["w"],
|
||||||
|
["w"],
|
||||||
|
["w"],
|
||||||
|
[],
|
||||||
|
[],
|
||||||
|
[],
|
||||||
|
[],
|
||||||
|
[],
|
||||||
|
[],
|
||||||
|
]
|
||||||
|
|
||||||
|
|
||||||
def test_lingbot_realtime_case_is_registered_by_default():
|
def test_lingbot_realtime_case_is_registered_by_default():
|
||||||
from sglang.multimodal_gen.test.server.gpu_cases import (
|
from sglang.multimodal_gen.test.server.gpu_cases import ONE_GPU_CASES
|
||||||
ONE_GPU_CASES,
|
|
||||||
_make_lingbot_realtime_plastic_beach_case,
|
|
||||||
)
|
|
||||||
|
|
||||||
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.id == "lingbot_world_realtime_plastic_beach"
|
||||||
assert case.run_consistency_check is True
|
assert case.run_consistency_check is True
|
||||||
assert case.run_perf_check is True
|
assert case.run_perf_check is True
|
||||||
assert case.sampling_params.realtime_output_format is None
|
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