refactor logprob processor layer (#20071)

2026-07-17 16:28:10 -07:00
parent 0ad0ff2e9e
commit 632adff9fd
7 changed files with 279 additions and 243 deletions
@@ -11,7 +11,7 @@ from sglang.srt.layers.dp_attention import (
    is_dp_attention_enabled,
 )
 from sglang.srt.layers.logits_processor import LogitsProcessorOutput
-from sglang.srt.layers.utils.logprob import get_token_ids_logprobs, get_top_logprobs
+from sglang.srt.layers.logprob_processor import get_token_ids_logprobs, get_top_logprobs
 from sglang.srt.runtime_context import get_parallel, get_server_args
 from sglang.srt.sampling.sampling_batch_info import SamplingBatchInfo
 from sglang.srt.sampling.sampling_params import TOP_K_ALL