Skip to content

Commit 660e35a

Browse files
committed
Merge branch 'main' into release/3.2
2 parents b8b39e7 + 492b85b commit 660e35a

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

swift/llm/infer/infer_engine/utils.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -86,7 +86,7 @@ def _get_response(self, response: str, is_finished: bool, token_len: int) -> str
8686
def get_printable_text(self, raw_tokens: List[int], is_finished: bool) -> str:
8787
raw_tokens = raw_tokens[self.cache_idx:]
8888
response = self.template.decode(
89-
raw_tokens, is_finished, tokenizer_kwargs=self.decode_kwargs, first_token=self.first_token)
89+
raw_tokens, is_finished=is_finished, tokenizer_kwargs=self.decode_kwargs, first_token=self.first_token)
9090
self.first_token = False
9191
response = self._align_blank_suffix(response)
9292
return self._get_response(response, is_finished, len(raw_tokens))

swift/llm/template/base.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -359,8 +359,8 @@ def decode_seq_cls(self, logits: torch.Tensor):
359359

360360
def decode(self,
361361
generate_ids: List[int],
362-
is_finished: bool = True,
363362
*,
363+
is_finished: bool = True,
364364
tokenizer_kwargs=None,
365365
first_token=True,
366366
**kwargs) -> Any:

0 commit comments

Comments
 (0)