torch.manual_seed(7)
model = ScoreModel().eval()
trace_input = torch.randn(2, 4)
check_inputs = [(torch.randn(1, 4),), (torch.randn(5, 4),)]
artifact = Path("score_model_torchscript.pt")