remove abundant contents
Browse files- config.json +1 -5
- 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": "
|
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+)
|
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'])]
|