Hi, is there anything regarding saving a PredictorSAM (segment anything) into the BentoML store? does this work?

bentoml.pytorch.save_model(
     "sam",  # Model name in the local Model Store
     trained_model,  # Model instance being saved
     signatures={   # Model signatures for Runner inference
         "__call__": {
             "batchable": True,
             "batch_dim": 0,
         },
     }
 )