jw2yang commited on
Commit
01ec8bb
·
1 Parent(s): 3c4a0e6

remove abundant contents

Browse files
Files changed (2) hide show
  1. config.json +1 -5
  2. modeling_magma.py +1 -1
config.json CHANGED
@@ -135,12 +135,11 @@
135
  "max_position_embeddings": 8192,
136
  "mm_hidden_size": 3072,
137
  "mm_projector_lr": null,
138
- "mm_projector_type": "mlp2x_gelu_segtokv9",
139
  "mm_use_im_patch_token": false,
140
  "mm_use_im_start_end": false,
141
  "mm_vision_select_feature": "patch",
142
  "mm_vision_select_layer": -2,
143
- "mm_vision_tower": "segtokv9_xxlarge",
144
  "num_attention_heads": 32,
145
  "num_hidden_layers": 32,
146
  "num_key_value_heads": 8,
@@ -149,9 +148,6 @@
149
  "rms_norm_eps": 1e-05,
150
  "rope_scaling": null,
151
  "rope_theta": 500000.0,
152
- "segtok_decoder_dim": 512,
153
- "segtok_num_queries": 201,
154
- "segtok_posembed": "sinusoidal",
155
  "tie_word_embeddings": false,
156
  "tokenizer_model_max_length": 4096,
157
  "tokenizer_padding_side": "right",
 
135
  "max_position_embeddings": 8192,
136
  "mm_hidden_size": 3072,
137
  "mm_projector_lr": null,
138
+ "mm_projector_type": "mlp2x_gelu",
139
  "mm_use_im_patch_token": false,
140
  "mm_use_im_start_end": false,
141
  "mm_vision_select_feature": "patch",
142
  "mm_vision_select_layer": -2,
 
143
  "num_attention_heads": 32,
144
  "num_hidden_layers": 32,
145
  "num_key_value_heads": 8,
 
148
  "rms_norm_eps": 1e-05,
149
  "rope_scaling": null,
150
  "rope_theta": 500000.0,
 
 
 
151
  "tie_word_embeddings": false,
152
  "tokenizer_model_max_length": 4096,
153
  "tokenizer_padding_side": "right",
modeling_magma.py CHANGED
@@ -96,7 +96,7 @@ class MagmaMultiModalProjector(nn.Module):
96
  vision_backbone = config.get('vision_backbone', 'convnextxxlarge')
97
  vision_backbone_size = vision_backbone.replace('convnext', '')
98
  projector_type = config.get('mm_projector_type', 'linear')
99
- mlp_gelu_match = re.match(r'^mlp(\d+)x_gelu_segtokv(\d+)$', projector_type)
100
  if mlp_gelu_match:
101
  mlp_depth = int(mlp_gelu_match.group(1))
102
  modules = [nn.Linear(config['mm_hidden_size'], config['hidden_size'])]
 
96
  vision_backbone = config.get('vision_backbone', 'convnextxxlarge')
97
  vision_backbone_size = vision_backbone.replace('convnext', '')
98
  projector_type = config.get('mm_projector_type', 'linear')
99
+ mlp_gelu_match = re.match(r'^mlp(\d+)x_gelu$', projector_type)
100
  if mlp_gelu_match:
101
  mlp_depth = int(mlp_gelu_match.group(1))
102
  modules = [nn.Linear(config['mm_hidden_size'], config['hidden_size'])]