ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update chat_templates.py

Browse source
This commit is contained in:
Daniel Han-Chen 2024-03-10 02:57:41 +11:00
commit 457b7ba6d6
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
unsloth/chat_templates.py
View file

@ -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