Refactor eagle bigram key matching (#13714)
This commit is contained in:
@@ -603,7 +603,7 @@ class PrefillAdder:
|
||||
req.prefix_indices = torch.cat([req.prefix_indices, new_indices])
|
||||
req.extend_input_len = len(req.fill_ids) - len(req.prefix_indices)
|
||||
prefix_len = len(req.prefix_indices)
|
||||
req.last_matched_prefix_len = prefix_len
|
||||
req.cache_protected_len = prefix_len
|
||||
|
||||
input_tokens = self.ceil_paged_tokens(req.extend_input_len)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user