model_type = r'vit_b' # model_ckpt_path = None model_ckpt_path = "checkpoints/sam_vit_b_01ec64.pth" device = 'cpu'