This commit is contained in:
@@ -997,6 +997,20 @@ STANDALONE_FILES = {
|
||||
],
|
||||
}
|
||||
|
||||
# test_update_weights_from_disk fails deterministically on ROCm: the diffusion
|
||||
# weight-update-from-disk reload path does not reconcile the diffusers
|
||||
# checkpoint layout with the sglang transformer parameters (shape mismatch ->
|
||||
# HTTP 400). Tracked in #31924. NVIDIA does not run standalone files, so this
|
||||
# test is AMD-only; skip it on ROCm until the reload path is fixed. Remove
|
||||
# this block (keeping the STANDALONE_FILES / est-time entry above) to
|
||||
# re-enable once #31924 lands.
|
||||
if current_platform.is_hip():
|
||||
STANDALONE_FILES["1-gpu"] = [
|
||||
f
|
||||
for f in STANDALONE_FILES["1-gpu"]
|
||||
if f != "../single_test_file/test_update_weights_from_disk.py"
|
||||
]
|
||||
|
||||
# New standalone files may omit an estimate once to learn the real CI runtime.
|
||||
# CI will use a fallback estimate for sharding, run the test, then print a
|
||||
# measured value that must be copied into STANDALONE_FILE_EST_TIMES.
|
||||
|
||||
Reference in New Issue
Block a user