Skip to content

Commit c7e1d58

Browse files
committed
fix bug
1 parent 8266ed7 commit c7e1d58

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

llm/server/server/triton_server.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -103,7 +103,7 @@ def _cache_special_tokens(self, batch_result):
103103
is_end = batch_result[i].get("is_end", 0)
104104
token_ids = batch_result[i]["token_ids"]
105105
return_all_tokens = batch_result[i].get("return_all_tokens", False)
106-
cache_special_token = False if is_end == 1 else (13 <= token_ids[0] <= 268)
106+
cache_special_token = False if is_end == 1 else True
107107
if is_end != 1 and (cache_special_token or return_all_tokens or self.cfg.disable_streaming):
108108
if batch_result[i]["req_id"] not in self.token_buffer:
109109
self.token_buffer[batch_result[i]["req_id"]] = list()

0 commit comments

Comments
 (0)