Update __init__.py

__init__.py

@@ -1,4 +1,8 @@
 # __init__.py
 from .modeling_custom import CustomConfig, CustomModel
+from transformers import AutoConfig, AutoModel, CONFIG_MAPPING, MODEL_MAPPING
 
-__all__ = ["CustomConfig", "CustomModel"]
+CONFIG_MAPPING.update({"custom_model": CustomConfig})
+MODEL_MAPPING.update({"custom_model": CustomModel})
+
+__all__ = ["CustomConfig", "CustomModel"]