Update configuration_mobilenet.py

configuration_mobilenet.py

@@ -1,7 +1,7 @@
 from transformers import PretrainedConfig
 
 class MobileNetV1Config(PretrainedConfig):
-    model_type = "mobilenet_v1_modern"
+    model_type = "mobilenet_custom"
 
     def __init__(
         self,
@@ -13,12 +13,5 @@ class MobileNetV1Config(PretrainedConfig):
         self.num_classes = num_classes
         self.block_dropout = block_dropout
         self.final_dropout = final_dropout
-        
-        # This mapping allows AutoModel to find the code when pushed to the Hub
-        if "auto_map" not in kwargs:
-            kwargs["auto_map"] = {
-                "AutoConfig": "configuration_mobilenet.MobileNetV1Config",
-                "AutoModelForImageClassification": "modeling_mobilenet.MobileNetV1ForImageClassification"
-            }
             
-        super().__init__(**kwargs)
+        super().__init__(**kwargs)