Fix DataCollatorForWholeWordMask again
# What does this PR do?

Fixes  #8388

I tested this change by running `run_mlm_wwm.py` (did not test the chinese version). Seems to work after also fixing the example. #8394

@sgugger 