How to have model defined in another file?

[stanleyshly]

In the github: it says to define the class like this
def init(self):
super().init()
self.encoder = nn.Sequential(nn.Linear(28 * 28, 128), nn.ReLU(), nn.Linear(128, 3))
self.decoder = nn.Sequential(nn.Linear(3, 128), nn.ReLU(), nn.Linear(128, 28 * 28))
, but is there anyway to import a model from another file like output=model(input), and use that in the class definition?