Add more statistics for spec decoding (#13317)

2025-11-15 14:20:28 +08:00
parent c2083116e3
commit 34851471b2
1 changed files with 3 additions and 0 deletions
--- a/python/sglang/srt/managers/tokenizer_manager.py
+++ b/python/sglang/srt/managers/tokenizer_manager.py
@@ -1779,6 +1779,9 @@ class TokenizerManager(TokenizerCommunicatorMixin):
                meta_info["spec_accept_length"] = (
                    recv_obj.completion_tokens[i] / recv_obj.spec_verify_ct[i]
                )
+                meta_info["spec_accept_token_num"] = accepted_tokens
+                meta_info["spec_draft_token_num"] = total_draft_tokens
+                meta_info["spec_verify_ct"] = recv_obj.spec_verify_ct[i]

    def _calculate_timing_metrics(
        self,