diff --git a/unsloth/chat_templates.py b/unsloth/chat_templates.py index 9de15b98cb..8bd7e62093 100644 --- a/unsloth/chat_templates.py +++ b/unsloth/chat_templates.py @@ -26,8 +26,8 @@ import shutil from .tokenizer_utils import ( load_correct_tokenizer, fix_sentencepiece_tokenizer, - patch_tokenizer, ) +from .models._utils import patch_tokenizer CHAT_TEMPLATES = {}