[diffusion] CI: fix CI test case skip problem (#15874)

Co-authored-by: Mick <mickjagger19@icloud.com>
This commit is contained in:
Prozac614
2025-12-26 19:42:20 +08:00
committed by GitHub
co-authored by Mick
parent 5d421db883
commit 3778c2fc6d
5 changed files with 44 additions and 10 deletions
@@ -16,6 +16,7 @@ from dataclasses import dataclass
from enum import Enum, auto
from typing import Any
from sglang.multimodal_gen import envs
from sglang.multimodal_gen.runtime.utils.logging_utils import init_logger
from sglang.multimodal_gen.utils import StoreBoolean
@@ -207,6 +208,10 @@ class SamplingParams:
if env_steps is not None and self.num_inference_steps is not None:
self.num_inference_steps = int(env_steps)
# Auto-enable stage logging if dump path is provided
if self.perf_dump_path:
envs.SGLANG_DIFFUSION_STAGE_LOGGING = True
def _validate(self):
"""
check if the sampling params is correct by itself