Spaces:

Rocketknight1
/

chat_template_creator

Running

Matt commited on Nov 22, 2023

Commit

ea3b994

•

1 Parent(s): 5005844

Bugfix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -31,7 +31,7 @@ def apply_chat_template(template, test_conversation, add_generation_prompt, clea
     pr_snippet = (
         "tokenizer = AutoTokenizer.from_pretrained(CHECKPOINT)",
         f"tokenizer.chat_template = \"{template}\"",
-        tokenizer.push_to_hub(CHECKPOINT, create_pr=True)
     )
     pr_snippet = "\n".join(pr_snippet)
     formatted = tokenizer.apply_chat_template(conversation, tokenize=False, add_generation_prompt=add_generation_prompt)

     pr_snippet = (
         "tokenizer = AutoTokenizer.from_pretrained(CHECKPOINT)",
         f"tokenizer.chat_template = \"{template}\"",
+        "tokenizer.push_to_hub(CHECKPOINT, create_pr=True)"
     )
     pr_snippet = "\n".join(pr_snippet)
     formatted = tokenizer.apply_chat_template(conversation, tokenize=False, add_generation_prompt=add_generation_prompt)