stevenbucaille commited on
Commit
ba1ed72
·
verified ·
1 Parent(s): 02a9f15
Files changed (2) hide show
  1. config.json +8 -4
  2. model.safetensors +2 -2
config.json CHANGED
@@ -1,11 +1,15 @@
1
  {
 
2
  "architectures": [
3
  "LightGlueForKeypointMatching"
4
  ],
5
- "attention_probs_dropout_prob": 0,
 
 
6
  "depth_confidence": 0.95,
7
  "descriptor_dim": 256,
8
  "filter_threshold": 0.1,
 
9
  "hidden_size": 256,
10
  "initializer_range": 0.02,
11
  "keypoint_detector_config": {
@@ -13,10 +17,10 @@
13
  },
14
  "model_type": "lightglue",
15
  "num_attention_heads": 4,
16
- "num_heads": 4,
17
- "num_layers": 9,
18
  "rotary_value": false,
19
  "torch_dtype": "float32",
20
- "transformers_version": "4.47.0.dev0",
21
  "width_confidence": 0.99
22
  }
 
1
  {
2
+ "_name_or_path": "lightglue/",
3
  "architectures": [
4
  "LightGlueForKeypointMatching"
5
  ],
6
+ "attention_bias": true,
7
+ "attention_dropout": 0.0,
8
+ "attention_probs_dropout_prob": 0.0,
9
  "depth_confidence": 0.95,
10
  "descriptor_dim": 256,
11
  "filter_threshold": 0.1,
12
+ "hidden_act": "gelu",
13
  "hidden_size": 256,
14
  "initializer_range": 0.02,
15
  "keypoint_detector_config": {
 
17
  },
18
  "model_type": "lightglue",
19
  "num_attention_heads": 4,
20
+ "num_hidden_layers": 9,
21
+ "num_key_value_heads": 4,
22
  "rotary_value": false,
23
  "torch_dtype": "float32",
24
+ "transformers_version": "4.49.0.dev0",
25
  "width_confidence": 0.99
26
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:87934e01272d482f500e5a820260958a768de30ced01c1146d5ecc4af5db38e4
3
- size 55020316
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:011c0b26ca1acc7d9361fa6e371a635a83549d2d4c92fdb7ee436acbc6257be0
3
+ size 55020320