[diffusion] chore: minor code cleanups and improve logging (#14916)

This commit is contained in:
Mick
2025-12-12 18:48:07 +08:00
committed by GitHub
parent 94e1251131
commit 82e33170e1
9 changed files with 40 additions and 78 deletions

View File

@@ -69,7 +69,7 @@ def run_pytest(files):
base_cmd = [sys.executable, "-m", "pytest", "-s", "-v", "--log-cli-level=INFO"]
max_retries = 2
max_retries = 4
# retry if the perf assertion failed, for {max_retries} times
for i in range(max_retries + 1):
cmd = list(base_cmd)

View File

@@ -461,7 +461,7 @@
"stages_ms": {
"InputValidationStage": 23,
"ImageEncodingStage": 1485.0,
"ImageVAEEncodingStage": 350.0,
"ImageVAEEncodingStage": 400.0,
"ConditioningStage": 0.13,
"TimestepPreparationStage": 13.78,
"LatentPreparationStage": 15.0,
@@ -1203,7 +1203,7 @@
"48": 3721.72,
"49": 3715.61
},
"expected_e2e_ms": 198187.89,
"expected_e2e_ms": 220000,
"expected_avg_denoise_ms": 3751.95,
"expected_median_denoise_ms": 3724.06
}