Update file model.py
Browse files
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):
|