diff --git a/mobile_sam/build_sam.py b/mobile_sam/build_sam.py
index 9a52c50..c9a51c8 100644
--- a/mobile_sam/build_sam.py
+++ b/mobile_sam/build_sam.py
@@ -88,7 +88,7 @@ def build_sam_vit_t(checkpoint=None):
     mobile_sam.eval()
     if checkpoint is not None:
         with open(checkpoint, "rb") as f:
-            state_dict = torch.load(f)
+            state_dict = torch.load(f, weights_only = False)
         mobile_sam.load_state_dict(state_dict)
     return mobile_sam