Fix TF template
# What does this PR do?

Fix the TF template.
@LysandreJik I think there is a problem with the Template test. It doesn't seem to take into account the changes in the current PR.<|||||>Good catch @jplu, thanks for fixing! Yes indeed, this test needs to be reworked. Tracking it in https://github.com/huggingface/transformers/issues/10065