Spaces:
Build error
Build error
File size: 513 Bytes
3b8cdb2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
from .blip import BLIPCaptioner
from .blip2 import BLIP2Captioner
from .git import GITCaptioner
from .base_captioner import BaseCaptioner
def build_captioner(type, device, args=None):
if type == 'blip':
return BLIPCaptioner(device, enable_filter=args.clip_filter)
elif type == 'blip2':
return BLIP2Captioner(device, enable_filter=args.clip_filter)
elif type == 'git':
return GITCaptioner(device, enable_filter=args.clip_filter)
else:
raise NotImplementedError("") |