Journey9ni commited on
Commit
33ff3a3
·
1 Parent(s): 63ddd96

Initial commit for new model

Browse files
Files changed (1) hide show
  1. config.json +25 -0
config.json CHANGED
@@ -250,5 +250,30 @@
250
  "torchscript": false,
251
  "typical_p": 1.0,
252
  "use_bfloat16": false
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
253
  }
254
  }
 
250
  "torchscript": false,
251
  "typical_p": 1.0,
252
  "use_bfloat16": false
253
+ },
254
+ "spatial_encoder_hidden_size": 768,
255
+ "num_attention_heads": 16,
256
+ "spatial_encoder_cfg": {
257
+ "architectures": [
258
+ "Mast3rEncoder"
259
+ ],
260
+ "pretrained_model_name_or_path": "../mast3r/checkpoints/MASt3R_ViTLarge_BaseDecoder_512_catmlpdpt_metric.pth",
261
+ "preprocessor_config": {
262
+ "do_convert_rgb": true,
263
+ "do_normalize": true,
264
+ "do_rescale": true,
265
+ "do_resize_and_center_crop": true,
266
+ "image_mean": [0.5, 0.5, 0.5],
267
+ "image_std": [0.5, 0.5, 0.5],
268
+ "resample": 3,
269
+ "rescale_factor": 0.00392156862745098,
270
+ "longest_edge": 512
271
+ }
272
+ },
273
+ "cross_attention_cfg": {
274
+ "architectures": [
275
+ "CrossAttention"
276
+ ],
277
+ "cross_attention_type": "cross_attn"
278
  }
279
  }