jiajunlong commited on
Commit
3a0a50a
1 Parent(s): 45a3074

Update configuration.py

Browse files
Files changed (1) hide show
  1. configuration.py +5 -1
configuration.py CHANGED
@@ -1,7 +1,11 @@
1
  from transformers import PretrainedConfig
2
  from transformers import CONFIG_MAPPING
3
  from transformers import AutoConfig
4
- from utils import *
 
 
 
 
5
 
6
  class TinyLlavaConfig(PretrainedConfig):
7
 
 
1
  from transformers import PretrainedConfig
2
  from transformers import CONFIG_MAPPING
3
  from transformers import AutoConfig
4
+
5
+ IGNORE_INDEX = -100
6
+ IMAGE_TOKEN_INDEX = -200
7
+ DEFAULT_IMAGE_TOKEN = "<image>"
8
+
9
 
10
  class TinyLlavaConfig(PretrainedConfig):
11