Spaces:

hujiecpp
/

PE3R

Running on Zero

hujiecpp commited on Feb 19

Commit

8c164d9

1 Parent(s): f875744

init project

Files changed (1) hide show

modules/pe3r/models.py CHANGED Viewed

@@ -18,7 +18,7 @@ class Models:
         # -- sam2 --
         # SAM2_CKP = "./checkpoints/sam2.1_hiera_large.pt"
-        SAM2_CKP = 'facebook/sam2.1-hiera-large'
         SAM2_CONFIG = "./configs/sam2.1/sam2.1_hiera_l.yaml"
         self.sam2 = build_sam2_video_predictor(SAM2_CONFIG, SAM2_CKP, device=device, apply_postprocessing=False)
         self.sam2.eval()

         # -- sam2 --
         # SAM2_CKP = "./checkpoints/sam2.1_hiera_large.pt"
+        SAM2_CKP = 'hujiecpp/sam2-1-hiera-large'
         SAM2_CONFIG = "./configs/sam2.1/sam2.1_hiera_l.yaml"
         self.sam2 = build_sam2_video_predictor(SAM2_CONFIG, SAM2_CKP, device=device, apply_postprocessing=False)
         self.sam2.eval()