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)
|