[diffusion] fix: fix hunyuanvideo and add 2-gpu ci test (#13720)

Co-authored-by: Mick <mickjagger19@icloud.com>
This commit is contained in:
Yuhao Yang
2025-11-22 20:54:51 +08:00
committed by GitHub
co-authored by Mick
parent ac43822634
commit a22de641ef
16 changed files with 822 additions and 438 deletions
@@ -164,7 +164,7 @@ def wait_for_req_perf_record(
request_id: str,
prev_len: int,
log_path: Path,
timeout: float = 300.0,
timeout: float = 30.0,
) -> tuple[RequestPerfRecord | None, int]:
"""
the stage metrics of this request should be in the performance_log file with {request-id}
@@ -173,7 +173,7 @@ def wait_for_req_perf_record(
deadline = time.time() + timeout
while time.time() < deadline:
records = read_perf_logs(log_path)
if len(records) == prev_len + 1:
if len(records) >= prev_len + 1:
# FIXME: unable to get rid from openai apis, this is a hack. we should compare rid
# potential error when there are multiple servers
return records[-1], len(records)
@@ -183,6 +183,8 @@ def wait_for_req_perf_record(
if os.environ.get("SGLANG_GEN_BASELINE", "0") == "1":
records = read_perf_logs(log_path)
return None, len(records)
logger.error(f"record: {records}")
raise AssertionError(f"Timeout waiting for stage metrics for request {request_id} ")