Upload model

config.json

@@ -7,6 +7,15 @@
     "AutoModelForObjectDetection": "modeling.LSPDetrModel"
   },
   "backbone": "microsoft/swinv2-tiny-patch4-window16-256",
+  "backbone_config": null,
+  "backbone_kwargs": {
+    "out_features": [
+      "stage1",
+      "stage2",
+      "stage3",
+      "stage4"
+    ]
+  },
   "depths": [
     6,
     2,
@@ -31,5 +40,7 @@
   ],
   "torch_dtype": "float32",
   "transformers_version": "4.51.3",
+  "use_pretrained_backbone": true,
+  "use_timm_backbone": false,
   "window_size": 16
 }

configuration.py

@@ -37,7 +37,11 @@ class LSPDetrConfig(PretrainedConfig):
             backbone_kwargs=backbone_kwargs,
         )
 
+        self.use_timm_backbone = use_timm_backbone
+        self.use_pretrained_backbone = use_pretrained_backbone
         self.backbone = backbone
+        self.backbone_config = backbone_config
+        self.backbone_kwargs = backbone_kwargs
         self.dim = dim
         self.num_classes = num_classes
         self.depths = depths