[diffusion] fix: fix stages not logged when perf_dump_path is provided (#16016)
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user