[diffusion] fix: fix stages not logged when perf_dump_path is provided (#16016)

This commit is contained in:
Mick
2025-12-28 23:17:43 +08:00
committed by GitHub
parent 7d02c8e59f
commit d7a3336ebe
6 changed files with 23 additions and 23 deletions
@@ -16,7 +16,6 @@ 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
@@ -208,10 +207,6 @@ 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