Update 2 files

- /model.py
- /trainer.py

model.py

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

trainer.py

@@ -10,7 +10,7 @@ class Trainer:
 
         #self.wandb = Wandb(config.wandb)
         
-        print(f"model config: {config.model}")
+        print(f"model config: {model}")
         print(f"config.params: {config.model.params}")
         #self.model = Model(config.model)