Apple MPS support?

[shersoni610]

Hello,

Does anyone knows how to use apple "MPS" instead of cpu/cuda device?

[JunkyByte]

Install torch with mps support then it should be enough to choose mps as backend

from mobile_sam import sam_model_registry, SamAutomaticMaskGenerator, SamPredictor

model_type = "vit_t"
sam_checkpoint = "./weights/mobile_sam.pt"

device = "mps"

mobile_sam = sam_model_registry[model_type](checkpoint=sam_checkpoint)
mobile_sam.to(device=device)
mobile_sam.eval()

predictor = SamPredictor(mobile_sam)
predictor.set_image(<your_image>)
masks, _, _ = predictor.predict(<input_prompts>)