Julian Bilcke
commited on
Commit
·
a73397c
1
Parent(s):
f577b1e
restore missing dependencies
Browse files- captioning_service.py +3 -0
captioning_service.py
CHANGED
|
@@ -11,6 +11,9 @@ from dataclasses import dataclass
|
|
| 11 |
from datetime import datetime
|
| 12 |
import cv2
|
| 13 |
import copy
|
|
|
|
|
|
|
|
|
|
| 14 |
from llava.constants import IMAGE_TOKEN_INDEX, DEFAULT_IMAGE_TOKEN, DEFAULT_IM_START_TOKEN, DEFAULT_IM_END_TOKEN
|
| 15 |
from llava.conversation import conv_templates, SeparatorStyle
|
| 16 |
|
|
|
|
| 11 |
from datetime import datetime
|
| 12 |
import cv2
|
| 13 |
import copy
|
| 14 |
+
|
| 15 |
+
from llava.model.builder import load_pretrained_model
|
| 16 |
+
from llava.mm_utils import tokenizer_image_token
|
| 17 |
from llava.constants import IMAGE_TOKEN_INDEX, DEFAULT_IMAGE_TOKEN, DEFAULT_IM_START_TOKEN, DEFAULT_IM_END_TOKEN
|
| 18 |
from llava.conversation import conv_templates, SeparatorStyle
|
| 19 |
|