From 457b7ba6d69c14fe5357f84d2164a0cc7166cded Mon Sep 17 00:00:00 2001 From: Daniel Han-Chen Date: Sun, 10 Mar 2024 02:57:41 +1100 Subject: [PATCH] Update chat_templates.py --- unsloth/chat_templates.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/unsloth/chat_templates.py b/unsloth/chat_templates.py index 715cf6b3c2..520c998c5c 100644 --- a/unsloth/chat_templates.py +++ b/unsloth/chat_templates.py @@ -257,9 +257,9 @@ def get_chat_template( assert("Unsloth: Can only map new tokens to EOS for now. Adding new tokens is not yet supported.") pass - if tokenizer.__class__.__name__.startswith("Gemma") and chat_template == "chatml": - chat_template = "gemma_chatml" - pass + # if tokenizer.__class__.__name__.startswith("Gemma") and chat_template == "chatml": + # chat_template = "gemma_chatml" + # pass old_padding_side = tokenizer.padding_side