Enable download tracking and update pipeline tag

README.md

@@ -1,5 +1,5 @@
 ---
-pipeline_tag: image-feature-extraction
+pipeline_tag: visual-document-retrieval
 library_name: pytorch
 license: mit
 tags:

config.json

@@ -0,0 +1,16 @@
+{
+  "model_type": "megaloc",
+  "architectures": ["MegaLoc"],
+  "feat_dim": 8448,
+  "num_clusters": 64,
+  "cluster_dim": 256,
+  "token_dim": 256,
+  "mlp_dim": 512,
+  "backbone": {
+    "type": "dinov2",
+    "embed_dim": 768,
+    "patch_size": 14,
+    "depth": 12,
+    "num_heads": 12
+  }
+}