flpelerin commited on
Commit
5a8f44b
·
1 Parent(s): b0e0141

Update file model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -0
model.py CHANGED
@@ -9,6 +9,7 @@ class Model:
9
  self.__dict__ = dict(config.__dict__)
10
 
11
  self.model = MambaLMHeadModel(MambaConfig(**self.params.__dict__)).to(GetDevice())
 
12
 
13
 
14
  def Log(self):
 
9
  self.__dict__ = dict(config.__dict__)
10
 
11
  self.model = MambaLMHeadModel(MambaConfig(**self.params.__dict__)).to(GetDevice())
12
+ self.Log()
13
 
14
 
15
  def Log(self):