Tangledup-ai/sam3_local
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

local

Browse Source
This commit is contained in:
quant
2026-02-15 12:25:24 +08:00
parent 2ec3c0711a
commit 8bb00ac928
2 changed files with 43 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
sam3/model_builder.py
View File

@@ -561,8 +561,8 @@ def build_sam3_image_model(
bpe_path=None,
device="cuda" if torch.cuda.is_available() else "cpu",
eval_mode=True,
checkpoint_path=None,
load_from_HF=True,
checkpoint_path="/home/quant/data/dev/sam3/sam3.pt",
load_from_HF=False,
enable_segmentation=True,
enable_inst_interactivity=False,
compile=False,
@@ -651,8 +651,8 @@ def download_ckpt_from_hf():
def build_sam3_video_model(
checkpoint_path: Optional[str] = None,
load_from_HF=True,
checkpoint_path: Optional[str] = "/home/quant/data/dev/sam3/sam3.pt",
load_from_HF=False,
bpe_path: Optional[str] = None,
has_presence_token: bool = True,
geo_encoder_use_img_cross_attn: bool = True,