Fix condition check for require_gathered_buffer (#15328)

2025-12-17 12:42:47 -08:00
parent 726fe3e72e
commit 5290cef97c
1 changed files with 1 additions and 1 deletions
--- a/python/sglang/srt/model_executor/model_runner.py
+++ b/python/sglang/srt/model_executor/model_runner.py
@@ -2783,7 +2783,7 @@ class ModelRunner:
        if (
            forward_batch.num_token_non_padded is not None
            and forward_batch.global_num_tokens_gpu is not None
-            and require_gathered_buffer
+            and require_gathered_buffer(self.server_args)
            and not is_nsa_enable_prefill_cp()
        ):
            forward_batch.adjust_num_token_non_padded_for_attn_tp(