From 2771e5042b45c09b0f5534dfb0f613a4dd30782f Mon Sep 17 00:00:00 2001 From: Daniel Han-Chen Date: Fri, 19 Apr 2024 03:36:45 +1000 Subject: [PATCH] Update _utils.py --- unsloth/models/_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py index 8864ebab4b..32da0a7345 100644 --- a/unsloth/models/_utils.py +++ b/unsloth/models/_utils.py @@ -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." )