Update 2 files

- /model.py
- /trainer.py

model.py

@@ -8,9 +8,9 @@ class Model:
     def __init__(self, config: Config):
         self.__dict__ = dict(config.__dict__)
 
-        print(f"params: {params}")
+        #print(f"params: {params}")
 
-        #self.model = MambaLMHeadModel(MambaConfig(params)).to(GetDevice())
+        self.model = MambaLMHeadModel(MambaConfig(self.params)).to(GetDevice())
 
 
     def AutoRegressiveLossFunction(self, input_ids, labels=None, criterion=None):

trainer.py

@@ -8,12 +8,12 @@ class Trainer:
     def __init__(self, config: Config):
         self.__dict__ = dict(config.__dict__)
 
-        print(f"self.dict: {self.__dict__}")
-        print(f"locals: {locals()}")
+        #print(f"self.dict: {self.__dict__}")
+        #print(f"locals: {locals()}")
 
         #self.wandb = Wandb(config.wandb)
         
-        print(f"model config: {model}")
-        print(f"config.params: {config.model.params}")
-        #self.model = Model(config.model)
+        #print(f"model config: {self.model}")
+        #print(f"config.params: {self.model.params}")
+        self.model = Model(config.model)