zlei6/ComfyUI
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI.git synced 2026-08-16 06:26:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: SAM (segment anything) 3.1 support (CORE-34) (#13408)

Browse Source
This commit is contained in:
Jukka Seppänen
2026-04-23 07:07:43 +03:00
committed by GitHub
parent e988df72f8
commit 749d5b4e8d
9 changed files with 3502 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
comfy/model_base.py
View File

@@ -54,6 +54,7 @@ import comfy.ldm.anima.model
import comfy.ldm.ace.ace_step15
import comfy.ldm.rt_detr.rtdetr_v4
import comfy.ldm.ernie.model
import comfy.ldm.sam3.detector
import comfy.model_management
import comfy.patcher_extension
@@ -1974,3 +1975,7 @@ class ErnieImage(BaseModel):
if cross_attn is not None:
out['c_crossattn'] = comfy.conds.CONDRegular(cross_attn)
return out
class SAM3(BaseModel):
def __init__(self, model_config, model_type=ModelType.FLOW, device=None):
super().__init__(model_config, model_type, device=device, unet_model=comfy.ldm.sam3.detector.SAM3Model)