danelcsb commited on
Commit
ceab007
·
verified ·
1 Parent(s): f852a05

Upload model

Browse files
Files changed (2) hide show
  1. config.json +224 -0
  2. model.safetensors +3 -0
config.json ADDED
@@ -0,0 +1,224 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "Sam2VideoModel"
4
+ ],
5
+ "binarize_mask_from_pts_for_mem_enc": true,
6
+ "enable_occlusion_spatial_embedding": false,
7
+ "enable_temporal_pos_encoding_for_object_pointers": true,
8
+ "fill_hole_area": 8,
9
+ "image_size": 1024,
10
+ "initializer_range": 0.02,
11
+ "mask_decoder_config": {
12
+ "attention_downsample_rate": 2,
13
+ "dynamic_multimask_stability_delta": 0.05,
14
+ "dynamic_multimask_stability_thresh": 0.98,
15
+ "dynamic_multimask_via_stability": true,
16
+ "feed_forward_hidden_act": "relu",
17
+ "hidden_act": "gelu",
18
+ "hidden_size": 256,
19
+ "iou_head_depth": 3,
20
+ "iou_head_hidden_dim": 256,
21
+ "mlp_dim": 2048,
22
+ "model_type": "",
23
+ "num_attention_heads": 8,
24
+ "num_hidden_layers": 2,
25
+ "num_multimask_outputs": 3,
26
+ "two_way_transformer_activation": "relu"
27
+ },
28
+ "mask_downsampler_embed_dim": 256,
29
+ "mask_downsampler_hidden_act": "gelu",
30
+ "mask_downsampler_kernel_size": 3,
31
+ "mask_downsampler_padding": 1,
32
+ "mask_downsampler_stride": 2,
33
+ "mask_downsampler_total_stride": 16,
34
+ "max_object_pointers_in_encoder": 16,
35
+ "memory_attention_apply_pe_at_cross_attn_keys": true,
36
+ "memory_attention_apply_pe_at_cross_attn_queries": false,
37
+ "memory_attention_apply_pe_at_self_attn": false,
38
+ "memory_attention_downsample_rate": 1,
39
+ "memory_attention_dropout": 0.1,
40
+ "memory_attention_feed_forward_hidden_act": "relu",
41
+ "memory_attention_feed_forward_hidden_size": 2048,
42
+ "memory_attention_hidden_size": 256,
43
+ "memory_attention_num_attention_heads": 1,
44
+ "memory_attention_num_layers": 4,
45
+ "memory_attention_rope_dropout": 0.1,
46
+ "memory_attention_rope_feat_sizes": [
47
+ 64,
48
+ 64
49
+ ],
50
+ "memory_attention_rope_theta": 10000,
51
+ "memory_encoder_hidden_size": 256,
52
+ "memory_encoder_output_channels": 64,
53
+ "memory_fuser_embed_dim": 256,
54
+ "memory_fuser_hidden_act": "gelu",
55
+ "memory_fuser_kernel_size": 7,
56
+ "memory_fuser_layer_scale_init_value": 1e-06,
57
+ "memory_fuser_num_layers": 2,
58
+ "memory_fuser_padding": 3,
59
+ "memory_fuser_use_depthwise_conv": true,
60
+ "model_type": "sam2",
61
+ "multimask_max_pt_num": 1,
62
+ "multimask_min_pt_num": 0,
63
+ "multimask_output_for_tracking": true,
64
+ "multimask_output_in_sam": true,
65
+ "non_overlap_masks": false,
66
+ "non_overlap_masks_for_mem_enc": false,
67
+ "num_maskmem": 7,
68
+ "preserve_temporal_direction_in_object_pointers": true,
69
+ "project_temporal_pos_encoding_in_object_pointers": false,
70
+ "prompt_encoder_config": {
71
+ "hidden_act": "gelu",
72
+ "hidden_size": 256,
73
+ "image_size": 1024,
74
+ "layer_norm_eps": 1e-06,
75
+ "mask_input_channels": 16,
76
+ "model_type": "",
77
+ "num_point_embeddings": 4,
78
+ "patch_size": 16,
79
+ "scale": 1
80
+ },
81
+ "sigmoid_bias_for_mem_enc": -10.0,
82
+ "sigmoid_scale_for_mem_enc": 20.0,
83
+ "torch_dtype": "float32",
84
+ "transformers_version": "4.54.0.dev0",
85
+ "vision_config": {
86
+ "backbone_channel_list": [
87
+ 896,
88
+ 448,
89
+ 224,
90
+ 112
91
+ ],
92
+ "backbone_config": {
93
+ "_name_or_path": "",
94
+ "add_cross_attention": false,
95
+ "architectures": null,
96
+ "bad_words_ids": null,
97
+ "begin_suppress_tokens": null,
98
+ "bos_token_id": null,
99
+ "chunk_size_feed_forward": 0,
100
+ "cross_attention_hidden_size": null,
101
+ "decoder_start_token_id": null,
102
+ "dim_mul": 2.0,
103
+ "diversity_penalty": 0.0,
104
+ "do_sample": false,
105
+ "drop_path_rate": 0.0,
106
+ "early_stopping": false,
107
+ "encoder_no_repeat_ngram_size": 0,
108
+ "eos_token_id": null,
109
+ "exponential_decay_length_penalty": null,
110
+ "finetuning_task": null,
111
+ "forced_bos_token_id": null,
112
+ "forced_eos_token_id": null,
113
+ "global_attention_blocks": [
114
+ 12,
115
+ 16,
116
+ 20
117
+ ],
118
+ "head_mul": 2.0,
119
+ "hidden_act": "gelu",
120
+ "hidden_size": 112,
121
+ "id2label": {
122
+ "0": "LABEL_0",
123
+ "1": "LABEL_1"
124
+ },
125
+ "image_size": 1024,
126
+ "initializer_range": 0.02,
127
+ "is_decoder": false,
128
+ "is_encoder_decoder": false,
129
+ "label2id": {
130
+ "LABEL_0": 0,
131
+ "LABEL_1": 1
132
+ },
133
+ "layer_norm_eps": 1e-06,
134
+ "length_penalty": 1.0,
135
+ "max_length": 20,
136
+ "min_length": 0,
137
+ "model_type": "sam2_hiera_det_model",
138
+ "no_repeat_ngram_size": 0,
139
+ "num_attention_heads": 2,
140
+ "num_beam_groups": 1,
141
+ "num_beams": 1,
142
+ "num_channels": 3,
143
+ "num_query_pool_stages": 3,
144
+ "num_return_sequences": 1,
145
+ "output_attentions": false,
146
+ "output_hidden_states": false,
147
+ "output_scores": false,
148
+ "pad_token_id": null,
149
+ "patch_kernel_size": 7,
150
+ "patch_padding": 3,
151
+ "patch_stride": 4,
152
+ "prefix": null,
153
+ "problem_type": null,
154
+ "pruned_heads": {},
155
+ "query_stride": [
156
+ 2,
157
+ 2
158
+ ],
159
+ "remove_invalid_values": false,
160
+ "repetition_penalty": 1.0,
161
+ "return_dict": true,
162
+ "return_dict_in_generate": false,
163
+ "sep_token_id": null,
164
+ "stages": [
165
+ 2,
166
+ 3,
167
+ 16,
168
+ 3
169
+ ],
170
+ "suppress_tokens": null,
171
+ "task_specific_params": null,
172
+ "temperature": 1.0,
173
+ "tf_legacy_loss": false,
174
+ "tie_encoder_decoder": false,
175
+ "tie_word_embeddings": true,
176
+ "tokenizer_class": null,
177
+ "top_k": 50,
178
+ "top_p": 1.0,
179
+ "torch_dtype": null,
180
+ "torchscript": false,
181
+ "typical_p": 1.0,
182
+ "use_bfloat16": false,
183
+ "window_positional_embedding_background_size": [
184
+ 14,
185
+ 14
186
+ ],
187
+ "window_spec": [
188
+ 8,
189
+ 4,
190
+ 14,
191
+ 7
192
+ ]
193
+ },
194
+ "backbone_feature_sizes": [
195
+ [
196
+ 256,
197
+ 256
198
+ ],
199
+ [
200
+ 128,
201
+ 128
202
+ ],
203
+ [
204
+ 64,
205
+ 64
206
+ ]
207
+ ],
208
+ "fpn_hidden_size": 256,
209
+ "fpn_interpolation_mode": "nearest",
210
+ "fpn_kernel_size": 1,
211
+ "fpn_padding": 0,
212
+ "fpn_stride": 1,
213
+ "fpn_top_down_levels": [
214
+ 2,
215
+ 3
216
+ ],
217
+ "fuse_type": "sum",
218
+ "hidden_act": "gelu",
219
+ "initializer_range": 0.02,
220
+ "layer_norm_eps": 1e-06,
221
+ "model_type": "sam2_vision_model",
222
+ "num_feature_levels": 3
223
+ }
224
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:28a635cd8178742499409efb7a549cb2540befe5c295bf5c48f1e138ebb6b186
3
+ size 323410168