From c0e6e2ebff88ef55ad4bda3864d82f193c5e79f6 Mon Sep 17 00:00:00 2001
From: Victor Umesiobi <vicksemmanuel58@gmail.com>
Date: Fri, 3 Jan 2025 04:49:05 +0000
Subject: [PATCH] minor changes

---
 src/latent_gemma/main.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/latent_gemma/main.py b/src/latent_gemma/main.py
index fef420a..610b703 100644
--- a/src/latent_gemma/main.py
+++ b/src/latent_gemma/main.py
@@ -35,8 +35,8 @@ class LatentReasoningGemmaForCausalLM(GemmaForCausalLM):
 
     }
     
-    def __init__(self, config_):
-        super().__init__(config_)
+    def __init__(self, config):
+        super().__init__(config)
         self.tokenizer: PreTrainedTokenizer = None
         self.current_stage = 0
         self.model_config = type(self).DEFAULT_CONFIG