LouisLi commited on
Commit
3d39b41
·
verified ·
1 Parent(s): dbb24d6

Update caption_anything/captioner/__init__.py

Browse files
caption_anything/captioner/__init__.py CHANGED
@@ -5,9 +5,9 @@ from .base_captioner import BaseCaptioner
5
 
6
 
7
  def build_captioner(type, device, args=None):
8
- if type == 'blip':
9
  return BLIPCaptioner(device, enable_filter=args.clip_filter)
10
- elif type == 'blip2':
11
  return BLIP2Captioner(device, enable_filter=args.clip_filter)
12
  elif type == 'git':
13
  return GITCaptioner(device, enable_filter=args.clip_filter)
 
5
 
6
 
7
  def build_captioner(type, device, args=None):
8
+ if type == 'blip2':
9
  return BLIPCaptioner(device, enable_filter=args.clip_filter)
10
+ elif type == 'blip':
11
  return BLIP2Captioner(device, enable_filter=args.clip_filter)
12
  elif type == 'git':
13
  return GITCaptioner(device, enable_filter=args.clip_filter)