tonic commited on
Commit
bc016c4
1 Parent(s): 9f3169c
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -7,11 +7,11 @@ import json
7
  from tqdm import tqdm
8
  import shortuuid
9
 
10
- from llava.constants import IMAGE_TOKEN_INDEX, DEFAULT_IMAGE_TOKEN, DEFAULT_IM_START_TOKEN, DEFAULT_IM_END_TOKEN
11
- from llava.conversation import conv_templates, SeparatorStyle
12
- from llava.model.builder import load_pretrained_model
13
- from llava.utils import disable_torch_init
14
- from llava.mm_utils import tokenizer_image_token, get_model_name_from_path, KeywordsStoppingCriteria
15
 
16
  from PIL import Image
17
  import math
 
7
  from tqdm import tqdm
8
  import shortuuid
9
 
10
+ from prometheusvl.constants import IMAGE_TOKEN_INDEX, DEFAULT_IMAGE_TOKEN, DEFAULT_IM_START_TOKEN, DEFAULT_IM_END_TOKEN
11
+ from prometheusvl.conversation import conv_templates, SeparatorStyle
12
+ from prometheusvl.model.builder import load_pretrained_model
13
+ from prometheusvl.utils import disable_torch_init
14
+ from prometheusvl.mm_utils import tokenizer_image_token, get_model_name_from_path, KeywordsStoppingCriteria
15
 
16
  from PIL import Image
17
  import math