[diffusion] CI: ad lingbot case (#27055)
This commit is contained in:
@@ -56,23 +56,6 @@ from sglang.multimodal_gen.test.test_utils import (
|
||||
_CACHE_DIT_CONFIG_DIR = Path(__file__).parent / "configs"
|
||||
|
||||
|
||||
def _make_lingbot_realtime_plastic_beach_case() -> DiffusionTestCase:
|
||||
return DiffusionTestCase(
|
||||
"lingbot_world_realtime_plastic_beach",
|
||||
DiffusionServerArgs(
|
||||
model_path="robbyant/lingbot-world-fast-diffusers",
|
||||
modality="video",
|
||||
num_gpus=1,
|
||||
extras=["--pipeline-class-name LingBotWorldCausalDMDPipeline"],
|
||||
text_encoder_cpu_offload=True,
|
||||
),
|
||||
LINGBOT_WORLD_REALTIME_sampling_params,
|
||||
run_component_accuracy_check=False,
|
||||
run_models_api_check=False,
|
||||
run_t2v_input_reference_check=False,
|
||||
)
|
||||
|
||||
|
||||
# All test cases with clean default values
|
||||
# To test different models, simply add more DiffusionCase entries
|
||||
ONE_GPU_CASES: list[DiffusionTestCase] = [
|
||||
@@ -415,6 +398,20 @@ ONE_GPU_CASES: list[DiffusionTestCase] = [
|
||||
),
|
||||
run_component_accuracy_check=False,
|
||||
),
|
||||
DiffusionTestCase(
|
||||
"lingbot_world_realtime_plastic_beach",
|
||||
DiffusionServerArgs(
|
||||
model_path="robbyant/lingbot-world-fast-diffusers",
|
||||
modality="video",
|
||||
num_gpus=1,
|
||||
extras=["--pipeline-class-name LingBotWorldCausalDMDPipeline"],
|
||||
text_encoder_cpu_offload=True,
|
||||
),
|
||||
LINGBOT_WORLD_REALTIME_sampling_params,
|
||||
run_component_accuracy_check=False,
|
||||
run_models_api_check=False,
|
||||
run_t2v_input_reference_check=False,
|
||||
),
|
||||
]
|
||||
|
||||
# Skip hunyuan3d on AMD: marching_cubes surface extraction produces invalid SDF on ROCm.
|
||||
@@ -762,7 +759,5 @@ if not current_platform.is_hip():
|
||||
)
|
||||
)
|
||||
|
||||
ONE_GPU_CASES.append(_make_lingbot_realtime_plastic_beach_case())
|
||||
|
||||
ONE_GPU_CASES += ONE_GPU_MODELOPT_FP8_CASES
|
||||
TWO_GPU_CASES = _with_default_num_gpus(TWO_GPU_CASES, 2)
|
||||
|
||||
@@ -33,7 +33,7 @@ if TYPE_CHECKING:
|
||||
|
||||
logger = init_logger(__name__)
|
||||
|
||||
SGL_TEST_FILES_CI_DATA_REVISION = "4a62a21f0f8bcc54c3bc6de5dbf25515bcb92b7e"
|
||||
SGL_TEST_FILES_CI_DATA_REVISION = "8dbe6bc44cda46065fde2168a3d10fd792e2b289"
|
||||
SGL_TEST_FILES_CONSISTENCY_GT_ROOT = (
|
||||
"https://raw.githubusercontent.com/"
|
||||
f"sgl-project/ci-data/{SGL_TEST_FILES_CI_DATA_REVISION}/"
|
||||
|
||||
Reference in New Issue
Block a user