HichTala
/

DiffusionDet

Object Detection

Model card Files Files and versions

HichTala commited on Apr 17

Commit

810cbf5

·

verified ·

1 Parent(s): a60ce1a

Upload config

Files changed (1) hide show

config.json +4 -4

config.json CHANGED Viewed

@@ -2,9 +2,10 @@
   "_name_or_path": "HichTala/DiffusionDet",
   "activation": "relu",
   "alpha": 0.25,
-  "architectures": [
-    "DiffusionDet"
-  ],
   "backbone": "resnet50",
   "backbone_config": null,
   "backbone_kwargs": {
@@ -243,7 +244,6 @@
     3
   ],
   "swin_size": "B",
-  "torch_dtype": "float32",
   "transformers_version": "4.49.0.dev0",
   "use_fed_loss": false,
   "use_focal": true,

   "_name_or_path": "HichTala/DiffusionDet",
   "activation": "relu",
   "alpha": 0.25,
+  "auto_map": {
+    "AutoConfig": "diffusiondet.configuration_diffusiondet.DiffusionDetConfig",
+    "AutoModelForObjectDetection": "diffusiondet.modeling_diffusiondet.DiffusionDet"
+  },
   "backbone": "resnet50",
   "backbone_config": null,
   "backbone_kwargs": {
     3
   ],
   "swin_size": "B",
   "transformers_version": "4.49.0.dev0",
   "use_fed_loss": false,
   "use_focal": true,