Update file model.py

model.py

@@ -13,8 +13,8 @@ class Model:
     def __init__(self, config: Config):
         self.__dict__ = dict(config.__dict__)
 
-        #self.model = MambaLMHeadModel(ModelArgs(**self.params.__dict__)).to(GetDevice())
-        self.model = Mamba(ModelConfig(**self.params.__dict__)).to(GetDevice())
+        #self.model = MambaLMHeadModel(MambaConfig(**self.params.__dict__)).to(GetDevice())
+        self.model = Mamba(ModelArgs(**self.params.__dict__)).to(GetDevice())
         self.log()