nodes/rdancer-ComfyUI_Florence2SAM2
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

use smaller model to improve CUDA usage

Browse Source
This commit is contained in:
SkalskiP
2024-08-01 14:58:49 +02:00
parent 576e22aaff
commit 5ae5bcae46
3 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
utils/sam.py
+2 -2
View File
@@ -7,8 +7,8 @@ from PIL import Image
from sam2.build_sam import build_sam2
from sam2.sam2_image_predictor import SAM2ImagePredictor
SAM_CHECKPOINT = "checkpoints/sam2_hiera_large.pt"
SAM_CONFIG = "sam2_hiera_l.yaml"
SAM_CHECKPOINT = "checkpoints/sam2_hiera_small.pt"
SAM_CONFIG = "sam2_hiera_s.yaml"
def load_sam_model(