From a64ef8878b0dfba1b3390b1211fb8bf28d50c06b Mon Sep 17 00:00:00 2001
From: Sami Jaghouar <sami.jaghouar@hotmail.fr>
Date: Sat, 28 Sep 2024 16:45:19 +0000
Subject: [PATCH] fix vocab size for debugmodel and real data

---
 src/zeroband/train.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/zeroband/train.py b/src/zeroband/train.py
index 4ff5134d..0d93d57a 100644
--- a/src/zeroband/train.py
+++ b/src/zeroband/train.py
@@ -92,6 +92,7 @@ def train(config: Config):
         num_workers=config.data.num_workers,
         fake_data=config.data.fake,
     )
+
     model, model_config = get_model(
         config.name_model,
         config.type_model,