torch.jit support

[stjbb]

Does your model not support torch.jit? I encounter an error when trying to convert the model with the following code. If there is a solution, please let me know.

import torch
from edge_sam import SamPredictor, sam_model_registry

sam_checkpoint = "./weights/edge_sam.pth"
model_type = "edge_sam"

# load model and run
sam = sam_model_registry[model_type](checkpoint=sam_checkpoint)
sam_jit = torch.jit.script(copy.deepcopy(sam))