Update custom_model.py

custom_model.py

@@ -4,7 +4,7 @@ from transformers import PreTrainedModel, PreTrainedTokenizer, AutoTokenizer, Au
 
 # Define a custom model class
 class CustomModel(PreTrainedModel):
-    config_class = None  # Set this to the custom configuration class if available
+    config_class = AutoConfig  # Use AutoConfig to dynamically load the configuration class
 
     def __init__(self, config):
         super().__init__(config)