Mangled tokenization with Llama 3.1 for string sequences containing<space>'m

[tomjorquera]

We observed that trying to tokenize/detokenize strings containing the sequence <space>'m would not give back the initial string, but would "eat" the leading whitespace.

For example, the string "for 'manual'" will be transformed into "for'manual'"

Investigating further, we also observed issue with strings containing <space>'s, making us think the issue may be related to trying to handle sequences such as "I'm".

System Info

transformers==4.46.2

Who can help?

I guess it's for @ArthurZucker and @itazap

Information

• The official example scripts
• My own modified scripts

Tasks

• An officially supported task in the examples folder (such as GLUE/SQuAD, ...)
• My own task or dataset (give details below)

Reproduction

Running:

from transformers import AutoTokenizer

prompt = """for 'manual'"""

tokenizer = AutoTokenizer.from_pretrained("meta-llama/Meta-Llama-3.1-8B-Instruct")
tokenizer.batch_decode(tokenizer([prompt])["input_ids"], skip_special_tokens=True)[0]

prints

"for'manual'"

(missing whitespace before the leading ')

Expected behavior

It should output the following

"for'manual'"

[tomjorquera]

Note: we encountered this issue first from TGI, as reported huggingface/text-generation-inference#2927

[ArthurZucker]

Hey! I think this is related to cleanup_tokenization_space that default to Truein the version of transformers that you are using

[tom-jorquera-pfx]

Thanks for the reply @ArthurZucker !

I can confirm that if I add clean_up_tokenization_spaces=False to the call to batch_decode, the issue disappears.

I'm not sure if this means this behavior is "normal" (or let's say expected at least), and what will be the consequences for setting this option to False.

I tested updating to v4.48.1 (actually I forgot to report I had already tested with v4.48.0 before, sorry), and the behavior is still the same. From the doc it seems indeed True is still the default in the latest version.

Is there something to improve here or shall will be left at that?