sam vit l

main.py

@@ -14,9 +14,12 @@ CORS(app)
 cudaOrNah = "cuda" if torch.cuda.is_available() else "cpu"
 print(cudaOrNah)
 
-# Global model setup
-checkpoint = "sam_vit_h_4b8939.pth"
-model_type = "vit_h"
+# Global model setup 
+# running out of memory adjusted
+# checkpoint = "sam_vit_h_4b8939.pth"
+# model_type = "vit_h"
+checkpoint = "sam_vit_l_0b3195.pth"
+model_type = "vit_l"
 sam = sam_model_registry[model_type](checkpoint=checkpoint)
 sam.to(device=cudaOrNah)
 mask_generator = SamAutomaticMaskGenerator(

sam_vit_l_0b3195.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3adcc4315b642a4d2101128f611684e8734c41232a17c648ed1693702a49a622
+size 1249524607