Update modeling_fake_custom.py

modeling_fake_custom.py

@@ -2,8 +2,10 @@ from transformers import GPT2Model, GPT2LMHeadModel
 from .configuration_fake_custom import FakeCustomConfig
 
 class FakeCustomModel(GPT2Model):
+    _auto_class = "AutoModel"
     config_class = FakeCustomConfig
 
 
 class FakeCustomModelForCausalLM(GPT2LMHeadModel):
+    _auto_class = "AutoModelForCausalLM"
     config_class = FakeCustomConfig