ttengwang
clean up code, add langchain for chatbox
9a84ec8
raw
history blame
297 Bytes
from .base_segmenter import BaseSegmenter
from caption_anything.utils.utils import seg_model_map
def build_segmenter(model_name, device, args=None, model=None):
return BaseSegmenter(device, args.segmenter_checkpoint, model_name, reuse_feature=not args.disable_reuse_features, model=model)