File size: 2,822 Bytes
c6be546
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
{
  "architectures": [
    "RepConTURModel"
  ],
  "auto_map": {
    "AutoConfig": "configuration_repcontur.RepConTURConfig",
    "AutoModel": "modeling_repcontur.RepConTURModel"
  },
  "ct_encoder": {
    "codebook_size": 8192,
    "dim": 512,
    "dim_head": 32,
    "encoder_type": "ctvit",
    "heads": 8,
    "image_size": 240,
    "patch_size": 20,
    "spatial_depth": 4,
    "temporal_depth": 4,
    "temporal_patch_size": 10
  },
  "ct_pooler": {
    "mlp_depth": 2,
    "n_heads": 8,
    "n_queries": 256
  },
  "decoder_layer": 4,
  "dtype": "bfloat16",
  "generation_kwargs": {},
  "img_context_token_id": 151667,
  "internvl_model_path": "OpenGVLab/InternVL2_5-1B",
  "llm_hidden_size": 896,
  "model_type": "repcontur",
  "num_heads": 32,
  "num_maskformer_stages": 5,
  "query_dim": 2048,
  "seg_token_idx": 151674,
  "special_tokens": [
    "[SEG]",
    "<p>",
    "</p>",
    "<vp>",
    "</vp>"
  ],
  "template": {
    "INSTRUCTION": "<|user|>\n{input}<|end|>\n<|assistant|>\n",
    "SEP": "\n",
    "STOP_WORDS": [
      "<|end|>"
    ],
    "SUFFIX": "<|end|>",
    "SUFFIX_AS_EOS": true,
    "SYSTEM": "<|system|>\n{system}<|end|>\n"
  },
  "text_embedding_dim": 2560,
  "transformers_version": "4.57.1",
  "vocab_size": 151679,
  "voxtell_arch_kwargs": {
    "conv_bias": true,
    "conv_op": "torch.nn.modules.conv.Conv3d",
    "dropout_op": null,
    "dropout_op_kwargs": null,
    "features_per_stage": [
      32,
      64,
      128,
      256,
      320,
      320
    ],
    "kernel_sizes": [
      [
        3,
        3,
        3
      ],
      [
        3,
        3,
        3
      ],
      [
        3,
        3,
        3
      ],
      [
        3,
        3,
        3
      ],
      [
        3,
        3,
        3
      ],
      [
        3,
        3,
        3
      ]
    ],
    "n_blocks_per_stage": [
      1,
      3,
      4,
      6,
      6,
      6
    ],
    "n_conv_per_stage_decoder": [
      1,
      1,
      1,
      1,
      1
    ],
    "n_stages": 6,
    "nonlin": "torch.nn.LeakyReLU",
    "nonlin_kwargs": {
      "inplace": true
    },
    "norm_op": "torch.nn.modules.instancenorm.InstanceNorm3d",
    "norm_op_kwargs": {
      "affine": true,
      "eps": 1e-05
    },
    "strides": [
      [
        1,
        1,
        1
      ],
      [
        2,
        2,
        2
      ],
      [
        2,
        2,
        2
      ],
      [
        2,
        2,
        2
      ],
      [
        2,
        2,
        2
      ],
      [
        2,
        2,
        2
      ]
    ]
  },
  "voxtell_decoder_layer": 4,
  "voxtell_kw_requires_import": [
    "conv_op",
    "norm_op",
    "dropout_op",
    "nonlin"
  ],
  "voxtell_num_heads": 32,
  "voxtell_num_maskformer_stages": 5,
  "voxtell_query_dim": 2048,
  "voxtell_text_embedding_dim": 2560
}