Skip to content

Commit 175ea7b

Browse files
wu6u3twgithub-actions[bot]anandhu-eng
authored
[Wan22] update video idx (#2530)
* update video idx * update * checking * update * fix sections * fix sections * fix sections * add * [Automated Commit] Format Codebase * run gh action * [Automated Commit] Format Codebase * per discussion on MLComm review * fix the sample.txt * fix checking path * promt-0.mp4, ids.mp4 * clean --------- Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com> Co-authored-by: ANANDHU S <71482562+anandhu-eng@users.noreply.github.com>
1 parent 2ee7190 commit 175ea7b

File tree

5 files changed

+71
-3
lines changed

5 files changed

+71
-3
lines changed

.github/workflows/auto-update-dev.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ name: Auto-Update Dev Branches from Master
33
on:
44
push:
55
branches:
6-
- master # Trigger workflow on commits to 'master' branch.
6+
- master # Trigger workflow on commits to 'master' branch
77
workflow_dispatch: {}
88

99
jobs:
Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,10 @@
1+
A boat sailing leisurely along the Seine River with the Eiffel Tower in background, pixel art-0.mp4, 130.mp4
2+
A panda drinking coffee in a cafe in Paris, watercolor painting-0.mp4, 106.mp4
3+
The bund Shanghai, black and white-0.mp4, 84.mp4
4+
an elephant spraying itself with water using its trunk to cool down-0.mp4, 59.mp4
5+
a car turning a corner-0.mp4, 12.mp4
6+
a truck anchored in a tranquil bay-0.mp4, 31.mp4
7+
The bund Shanghai, in cyberpunk style-0.mp4, 86.mp4
8+
Gwen Stacy reading a book, in cyberpunk style-0.mp4, 122.mp4
9+
skyscraper-0.mp4, 223.mp4
10+
a shark is swimming in the ocean, animated style-0.mp4, 96.mp4

tools/submission/submission_checker/checks/accuracy_check.py

Lines changed: 36 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
from ..constants import *
33
from ..loader import SubmissionLogs
44
from ..configuration.configuration import Config
5+
from ..utils import check_extra_files
56
import re
67
import os
78

@@ -25,6 +26,9 @@ class AccuracyCheck(BaseCheck):
2526
- `loadgen_errors_check`: Fails if Loadgen reported non-ignored errors.
2627
- `dataset_check`: Verifies the reported sample count matches the
2728
configured dataset size unless the check is skipped.
29+
- `extra_files_check`: For benchmarks in REQUIRED_ACC_BENCHMARK (e.g.
30+
stable-diffusion-xl, wan-2.2-t2v-a14b), verifies required extra
31+
artifacts (e.g. images/, videos/) exist in the accuracy directory.
2832
2933
Attributes:
3034
submission_logs (SubmissionLogs): Holder for submission log paths
@@ -78,6 +82,7 @@ def setup_checks(self):
7882
self.checks.append(self.accuracy_json_check)
7983
self.checks.append(self.loadgen_errors_check)
8084
self.checks.append(self.dataset_check)
85+
self.checks.append(self.extra_files_check)
8186

8287
def accuracy_result_check(self):
8388
"""Validate reported accuracy metrics in `accuracy.txt`.
@@ -234,3 +239,34 @@ def dataset_check(self):
234239
)
235240
return False
236241
return True
242+
243+
def extra_files_check(self):
244+
"""Verify required extra accuracy files for certain benchmarks.
245+
246+
For models in REQUIRED_ACC_BENCHMARK (e.g. stable-diffusion-xl
247+
images, wan-2.2-t2v-a14b videos), ensures the accuracy directory
248+
contains the required subdirs and files. Skipped if
249+
skip_extra_accuracy_files_check is set.
250+
251+
Returns:
252+
bool: True if the check is skipped, the model has no extra
253+
requirements, or all required files exist; False otherwise.
254+
"""
255+
if self.config.skip_extra_accuracy_files_check:
256+
return True
257+
if self.model not in REQUIRED_ACC_BENCHMARK:
258+
return True
259+
if self.config.version not in REQUIRED_ACC_BENCHMARK[self.model]:
260+
return True
261+
acc_dir = os.path.dirname(self.path)
262+
target_files = REQUIRED_ACC_BENCHMARK[self.model][self.config.version]
263+
extra_files_pass, missing_files = check_extra_files(
264+
acc_dir, target_files)
265+
if not extra_files_pass:
266+
self.log.error(
267+
"%s expected to have the following extra files (%s)",
268+
acc_dir,
269+
missing_files,
270+
)
271+
return False
272+
return True

tools/submission/submission_checker/constants.py

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1123,6 +1123,22 @@
11231123
"2289",
11241124
]
11251125
},
1126+
},
1127+
"wan-2.2-t2v-a14b": {
1128+
"v6.0": {
1129+
"videos": [
1130+
"130",
1131+
"106",
1132+
"84",
1133+
"59",
1134+
"12",
1135+
"31",
1136+
"86",
1137+
"122",
1138+
"233",
1139+
"96",
1140+
]
1141+
},
11261142
}
11271143
}
11281144
REQUIRED_MEASURE_FILES = ["user.conf", "README.md"]
@@ -1701,6 +1717,7 @@
17011717
"v6.0": "{division}/{submitter}/results/{system}/{benchmark}/{scenario}/TEST08/verify_accuracy.txt",
17021718
"default": "{division}/{submitter}/results/{system}/{benchmark}/{scenario}/TEST08/verify_accuracy.txt",
17031719
}
1720+
17041721
TEST07_ACC_PATH = {
17051722
"v6.0": "{division}/{submitter}/results/{system}/{benchmark}/{scenario}/TEST07/verify_accuracy.txt",
17061723
"default": "{division}/{submitter}/results/{system}/{benchmark}/{scenario}/TEST07/verify_accuracy.txt",

tools/submission/submission_checker/utils.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -57,8 +57,13 @@ def check_extra_files(path, target_files):
5757
for target_file in target_files[dir]:
5858
if target_file not in files:
5959
check_pass = False
60-
missing_files.append(
61-
f"{os.path.join(path, dir, target_file)}.png")
60+
if "images" in dir:
61+
missing_files.append(
62+
f"{os.path.join(path, dir, target_file)}.png")
63+
if "videos" in dir:
64+
missing_files.append(
65+
f"{os.path.join(path, dir, target_file)}.mp4")
66+
6267
if "captions" not in files:
6368
missing_files.append(
6469
f"{os.path.join(path, dir, 'captions.txt')}")

0 commit comments

Comments
 (0)