Fix GPT2 test

Rocketknight1 · Rocketknight1 · commit 7f5987d8c82f · 2024-07-22T16:07:59.000+01:00
diff --git a/tests/models/gpt2/test_tokenization_gpt2.py b/tests/models/gpt2/test_tokenization_gpt2.py
@@ -280,6 +280,7 @@ def test_special_tokens_mask_input_pairs_and_bos_token(self):
     @require_jinja
     def test_tokenization_for_chat(self):
         tokenizer = GPT2Tokenizer.from_pretrained(self.tmpdirname)
+        tokenizer.chat_template = '{% for message in messages %}{{ message.content }}{{ eos_token }}{% endfor %}'
         test_chats = [
             [{"role": "system", "content": "You are a helpful chatbot."}, {"role": "user", "content": "Hello!"}],
             [

Original file line number	Diff line number	Diff line change
`@@ -280,6 +280,7 @@ def test_special_tokens_mask_input_pairs_and_bos_token(self):`
`280`	`280`	`@require_jinja`
`281`	`281`	`def test_tokenization_for_chat(self):`
`282`	`282`	`tokenizer = GPT2Tokenizer.from_pretrained(self.tmpdirname)`
	`283`	`+ tokenizer.chat_template = '{% for message in messages %}{{ message.content }}{{ eos_token }}{% endfor %}'`
`283`	`284`	`test_chats = [`
`284`	`285`	`[{"role": "system", "content": "You are a helpful chatbot."}, {"role": "user", "content": "Hello!"}],`
`285`	`286`	`[`