Skip to content

Commit bf89017

Browse files
committed
fix vllm_to_hf_output
1 parent 50e9c2d commit bf89017

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/models/test_fuyu.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ def vllm_to_hf_output(vllm_output: Tuple[List[int], str,
2525
"""Sanitize vllm output to be comparable with hf output."""
2626
output_ids, output_str, out_logprobs = vllm_output
2727

28-
hf_output_str = output_str + "|ENDOFTEXT|"
28+
hf_output_str = output_str.lstrip() + "|ENDOFTEXT|"
2929

3030
return output_ids, hf_output_str, out_logprobs
3131

0 commit comments

Comments
 (0)