:construction: wip

model.py

@@ -1,3 +1,4 @@
+from pathlib import Path
 from typing import Tuple
 import torch
 from torch import nn
@@ -59,7 +60,23 @@ class Autoencoder(nn.Module):
 
 
 class MessageModel:
+
     def __init__(self, msg='hello, world'):
         self.msg = msg
+
     def __call__(self):
         print(self.msg)
+
+    @classmethod
+    def from_pretrained(cls, path):
+        path = Path(path)
+        msg_file_path = path / 'message.txt'
+        assert msg_file_path.exists()
+        msg = msg_file_path.read_text()
+        return cls(msg)
+
+    def save_pretrained(self, path):
+        path = Path(path)
+        path.mkdir(exist_ok=True, parents=True)
+        msg_file_path = path / 'message.txt'
+        msg_file_path.write_text(self.msg)