Spaces:

hujiecpp
/

PE3R

Running on Zero

hujiecpp commited on Feb 19

Commit

ae113d0

1 Parent(s): 562c140

init project

Files changed (1) hide show

modules/pe3r/models.py CHANGED Viewed

@@ -28,7 +28,7 @@ class Models:
         # -- mobilesamv2 & sam1 --
         # SAM1_ENCODER_CKP = './checkpoints/sam_vit_h.pt'
-        SAM1_ENCODER_CKP = 'facebook/sam-vit-huge'
         SAM1_DECODER_CKP = './checkpoints/Prompt_guided_Mask_Decoder.pt'
         self.mobilesamv2 = sam_model_registry['sam_vit_h'](None)
         image_encoder=sam_model_registry['sam_vit_h_encoder'](SAM1_ENCODER_CKP)

         # -- mobilesamv2 & sam1 --
         # SAM1_ENCODER_CKP = './checkpoints/sam_vit_h.pt'
+        SAM1_ENCODER_CKP = 'facebook/sam-vit-huge/model.safetensors'
         SAM1_DECODER_CKP = './checkpoints/Prompt_guided_Mask_Decoder.pt'
         self.mobilesamv2 = sam_model_registry['sam_vit_h'](None)
         image_encoder=sam_model_registry['sam_vit_h_encoder'](SAM1_ENCODER_CKP)