HichTala commited on
Commit
a2f3955
·
verified ·
1 Parent(s): 92bb84d

Upload configuration_diffusiondet.py

Browse files
Files changed (1) hide show
  1. configuration_diffusiondet.py +1 -0
configuration_diffusiondet.py CHANGED
@@ -92,6 +92,7 @@ class DiffusionDetConfig(PretrainedConfig):
92
  # Auto mapping
93
  self.auto_map = {
94
  "AutoConfig": "configuration_diffusiondet.DiffusionDetConfig",
 
95
  "AutoModelForObjectDetection": "modeling_diffusiondet.DiffusionDet"
96
  }
97
 
 
92
  # Auto mapping
93
  self.auto_map = {
94
  "AutoConfig": "configuration_diffusiondet.DiffusionDetConfig",
95
+ "AutoImageProcessor": "image_processing_diffusiondet.DiffusionDetImageProcessor",
96
  "AutoModelForObjectDetection": "modeling_diffusiondet.DiffusionDet"
97
  }
98