Skip to content

Commit 8bfb67e

Browse files
committed
fix ut
1 parent ef6e6d7 commit 8bfb67e

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

tests/output/test_process_batch_output.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -169,6 +169,7 @@ def setup_token_processor(self, speculative_decoding=False, use_logprobs=False):
169169
processor.accept_token_num_per_head_per_request = {}
170170
processor.accept_token_num_per_head = [0] * MAX_DRAFT_TOKENS
171171
processor.use_sampling_mask = False
172+
processor._benchmark_logger = None
172173

173174
# processor._recycle_resources = Mock()
174175

0 commit comments

Comments
 (0)