Spaces:
Running
Running
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 == '
|
9 |
return BLIPCaptioner(device, enable_filter=args.clip_filter)
|
10 |
-
elif type == '
|
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)
|