jbilcke-hf HF Staff commited on
Commit
a73397c
·
1 Parent(s): f577b1e

restore missing dependencies

Browse files
Files changed (1) hide show
  1. 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