Update _utils.py

This commit is contained in:
Daniel Han-Chen 2024-04-19 03:36:45 +10:00
commit 2771e5042b

View file

@ -153,7 +153,7 @@ def patch_tokenizer(model, tokenizer):
tokenizer.pad_token = tokenizer.unk_token
else:
name = model.config._name_or_path if model is not None else "Model"
logger.warning_one(
logger.warning_once(
f"{name} does not have a padding or unknown token!\n"\
f"Will use the EOS token of id {tokenizer.eos_token_id} as padding."
)