Add model

config.json

@@ -12,31 +12,28 @@
   "initializer_range": 0.02,
   "intermediate_size": 512,
   "keypoint_detector_config": {
-    "border_removal_distance": 4,
-    "decoder_hidden_size": 256,
+    "_name_or_path": "stevenbucaille/disk",
+    "architectures": [
+      "DiskForKeypointDetection"
+    ],
+    "auto_map": {
+      "AutoConfig": "configuration_disk.DiskConfig",
+      "AutoModelForKeypointDetection": "modeling_disk.DiskForKeypointDetection"
+    },
     "descriptor_decoder_dim": 128,
     "detection_threshold": 0.0,
-    "encoder_hidden_sizes": [
-      64,
-      64,
-      128,
-      128
-    ],
-    "initializer_range": 0.02,
-    "keypoint_decoder_dim": 65,
-    "keypoint_threshold": 0.005,
-    "max_keypoints": -1,
     "max_num_keypoints": null,
-    "model_type": "superpoint",
-    "nms_radius": 4,
+    "model_type": "disk",
     "nms_window_size": 5,
     "pad_if_not_divisible": true,
+    "torch_dtype": "float32",
     "weights": "depth"
   },
   "model_type": "lightglue",
   "num_attention_heads": 4,
   "num_hidden_layers": 9,
   "num_key_value_heads": 4,
+  "torch_dtype": "float32",
   "transformers_version": "4.54.0.dev0",
   "width_confidence": 0.99
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5bc5591a425692cbb52413fb2b735b6dce4e71d9cd53438990309d445b6e8fce
-size 54317792
+oid sha256:dd831d7372c85bc8b7aa1a5e17d6aee20fb13049d3aaf6fc3753734bdf1dd02b
+size 54314488