From 313453cae69791d7650cda8f0134d4233f6f253c Mon Sep 17 00:00:00 2001
From: Gustavo de Rosa <gugarosa@users.noreply.huggingface.co>
Date: Sat, 27 Jan 2024 17:21:23 +0000
Subject: [PATCH] Add EOS token to the generation config (#101)

- Add EOS token to the generation config (9a5117d15e407c48fa40f8adbac1244abae52d6f)


Co-authored-by: Joao Gante <joaogante@users.noreply.huggingface.co>
---
 generation_config.json | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/generation_config.json b/generation_config.json
index d0015bd..cf4ba0e 100644
--- a/generation_config.json
+++ b/generation_config.json
@@ -1,4 +1,6 @@
 {
   "_from_model_config": true,
-  "transformers_version": "4.37.0.dev0"
+  "transformers_version": "4.37.0.dev0",
+  "eos_token_id": 50256,
+  "bos_token_id": 50256
 }