config.json

{  
  "model_type": "llama",  
  "auto_map": {  
    "AutoModelForCausalLM": "test_module.MaliciousModel",  
    "AutoConfig": "test_module.MaliciousModel"  
  },  
  "architectures": ["MaliciousModel"]  
}