[CI] Revert ModelOpt NVFP4 threshold relax (#29844)

This commit is contained in:
Xiaoyu Zhang
2026-07-04 21:17:32 +08:00
committed by GitHub
parent 36fc0093d6
commit 6dd0cefb2a
3 changed files with 29 additions and 12 deletions
@@ -352,6 +352,8 @@ jobs:
- name: Install dependencies
timeout-minutes: 20
env:
SGLANG_CI_EARLY_LD_LIBRARY_PATH: "1"
run: |
CUSTOM_BUILD_SGL_KERNEL=${{inputs.sgl_kernel}} bash scripts/ci/cuda/ci_install_dependency.sh diffusion
@@ -19,12 +19,6 @@
"psnr_threshold": 28.0,
"mean_abs_diff_threshold": 6.3
},
"flux2_modelopt_nvfp4_t2i": {
"clip_threshold": 0.98,
"ssim_threshold": 0.95,
"psnr_threshold": 17.0,
"mean_abs_diff_threshold": 4.0
},
"ideogram4_nvfp4_t2i": {
"clip_threshold": 0.98,
"ssim_threshold": 0.95,
@@ -163,12 +157,6 @@
"psnr_threshold": 27.1,
"mean_abs_diff_threshold": 7.2
},
"wan22_modelopt_nvfp4_t2v": {
"clip_threshold": 0.98,
"ssim_threshold": 0.95,
"psnr_threshold": 24.0,
"mean_abs_diff_threshold": 4.0
},
"fastwan2_2_ti2v_5b": {
"clip_threshold": 0.97,
"ssim_threshold": 0.92,
+27
View File
@@ -196,6 +196,27 @@ setup_pip_toolchain() {
mark_step_done "${FUNCNAME[0]}"
}
remove_stale_cuda12_nvidia_wheels() {
if [ "$CU_MAJOR" != "13" ]; then
mark_step_done "${FUNCNAME[0]}"
return
fi
mapfile -t STALE_CUDA12_NVIDIA_WHEELS < <(
python3 -m pip list --format=freeze | sed -n 's/^\(nvidia-.*-cu12\)==.*/\1/p'
)
if [ ${#STALE_CUDA12_NVIDIA_WHEELS[@]} -eq 0 ]; then
echo "No stale CUDA 12 NVIDIA wheels found for ${CU_VERSION} job"
mark_step_done "${FUNCNAME[0]}"
return
fi
echo "Removing stale CUDA 12 NVIDIA wheels from ${CU_VERSION} job: ${STALE_CUDA12_NVIDIA_WHEELS[*]}"
$PIP_UNINSTALL_CMD "${STALE_CUDA12_NVIDIA_WHEELS[@]}" $PIP_UNINSTALL_SUFFIX
mark_step_done "${FUNCNAME[0]}"
}
uninstall_stale_flashinfer() {
# Keep flashinfer packages if version matches to avoid re-downloading:
# - flashinfer-cubin: 150+ MB
@@ -552,8 +573,14 @@ main() {
install_apt_packages
clean_site_packages
setup_pip_toolchain
remove_stale_cuda12_nvidia_wheels
uninstall_stale_flashinfer
install_sglang
# Diffusion B200 CI imports torch inside install_sglang_kernel after removing
# stale CUDA 12 NVIDIA wheels, so opt into one early LD_LIBRARY_PATH refresh.
if [ "${SGLANG_CI_EARLY_LD_LIBRARY_PATH:-0}" = "1" ]; then
setup_ld_library_path
fi
install_sglang_kernel
install_sglang_router
download_flashinfer_cache