Skip to content

Commit 47a8296

Browse files
committed
Fix style
Signed-off-by: SimJeg <sjegou@nvidia.com>
1 parent 8dd4622 commit 47a8296

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

kvpress/pipeline.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -311,7 +311,7 @@ def generate_answer(
311311
generated_ids.append(new_id)
312312
if new_id.item() in should_stop_token_ids:
313313
break
314-
answer = self.tokenizer.decode(torch.stack(generated_ids), skip_special_tokens=True)
314+
answer = str(self.tokenizer.decode(torch.stack(generated_ids), skip_special_tokens=True))
315315
return answer
316316

317317
def postprocess(self, model_outputs, single_question):

0 commit comments

Comments
 (0)