TongkunGuan commited on
Commit
e38d4bf
·
verified ·
1 Parent(s): a416baf

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -6
README.md CHANGED
@@ -106,7 +106,7 @@ from transformers import AutoTokenizer
106
  from internvl.model.internvl_chat import InternVLChatConfig, InternVisionModel
107
  from utils import post_process, generate_similiarity_map, load_model, load_image
108
 
109
- checkpoint = '/mnt/dolphinfs/ssd_pool/docker/user/hadoop-mt-ocr/guantongkun/work_dirs/8B_all_32/checkpoint-370000'
110
  image_path = './demo_images/0000000.png'
111
  input_query = '11/12/2020'
112
  out_dir = 'results'
@@ -146,11 +146,6 @@ attn_map = similarity.reshape(len(input_embedings), resized_size[0], resized_siz
146
 
147
  """generate map locally """
148
  generate_similiarity_map(images, attn_map, target_aspect_ratio, all_bpe_strings, out_dir)
149
-
150
-
151
- """user command """
152
- # python quick_start.py
153
-
154
  ```
155
 
156
  ### Evaluation on Vision Capability
 
106
  from internvl.model.internvl_chat import InternVLChatConfig, InternVisionModel
107
  from utils import post_process, generate_similiarity_map, load_model, load_image
108
 
109
+ checkpoint = 'xxx/checkpoint-370000'
110
  image_path = './demo_images/0000000.png'
111
  input_query = '11/12/2020'
112
  out_dir = 'results'
 
146
 
147
  """generate map locally """
148
  generate_similiarity_map(images, attn_map, target_aspect_ratio, all_bpe_strings, out_dir)
 
 
 
 
 
149
  ```
150
 
151
  ### Evaluation on Vision Capability