jw2yang commited on
Commit
b33355b
·
verified ·
1 Parent(s): 4446406

change image_token_index to image_token_id

Browse files
Files changed (1) hide show
  1. configuration_magma.py +3 -3
configuration_magma.py CHANGED
@@ -108,19 +108,19 @@ class MagmaConfig(PretrainedConfig):
108
  >>> configuration = model.config
109
  ```"""
110
 
111
- model_type = "magma"
112
  keys_to_ignore_at_inference = ["past_key_values"]
113
 
114
  def __init__(
115
  self,
116
  vision_config=None,
117
  text_config=None,
118
- image_token_index=None,
119
  tie_word_embeddings=False,
120
  **kwargs,
121
  ):
122
  self.vision_config = vision_config
123
- self.image_token_index = image_token_index
124
 
125
  if isinstance(text_config, dict):
126
  text_config["model_type"] = text_config["model_type"] if "model_type" in text_config else "llama"
 
108
  >>> configuration = model.config
109
  ```"""
110
 
111
+ model_type = "magma"
112
  keys_to_ignore_at_inference = ["past_key_values"]
113
 
114
  def __init__(
115
  self,
116
  vision_config=None,
117
  text_config=None,
118
+ image_token_id=None,
119
  tie_word_embeddings=False,
120
  **kwargs,
121
  ):
122
  self.vision_config = vision_config
123
+ self.image_token_index = image_token_id
124
 
125
  if isinstance(text_config, dict):
126
  text_config["model_type"] = text_config["model_type"] if "model_type" in text_config else "llama"