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'])]
|