Is it possible to bake custom torchscript classes into scripted models?

deployment
1

Suppose we have a torch model like this

from script_class import CustomClass
class Model(nn.Module):
    def __init__(self):
        super(Model, self).__init___()
        self.object = CustomClass()
   def forward(self):
       print('hello')

Then i script the model and save it as ‘model.pt’
Next when I load it without import CustomClass and run it, it will crash with unreadable c++ errors

How do I bake the class into the scripted model, something like torch.jit.export but for data structures?