huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-04-19 22:02:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(server): Use cleanup_tokenization_spaces=False for lossless decoding (#13)

Browse Source 
Fixes #12 in the easiest way I could think of.
This commit is contained in:
Nicolas Patry 2023-01-03 11:07:05 +01:00 committed by GitHub
parent 60472f9d2b
commit b94f30215f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
server/text_generation/models/causal_lm.py
View File

@ -354,7 +354,8 @@ class CausalLM(Model):
if stop: if stop:
# Decode all tokens # Decode all tokens
output_text = self.tokenizer.decode( output_text = self.tokenizer.decode(
all_input_ids.squeeze(-1), skip_special_tokens=True all_input_ids.squeeze(-1), skip_special_tokens=True,
cleanup_tokenization_spaces=False
) )
# Slice with input_length to remove padding # Slice with input_length to remove padding
token_ids = all_input_ids[-new_input_length:] token_ids = all_input_ids[-new_input_length:]