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.
|
# 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
|
# 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.
|
# measured value that must be copied into STANDALONE_FILE_EST_TIMES.
|
||||||
|
|||||||
Reference in New Issue
Block a user