Update config.json

config.json

@@ -34,6 +34,7 @@
   "vocab_size": 128256,
   "auto_map": {     
 	"AutoConfig": "configuration_ilama.IlamaConfig",     
-	"AutoModel": "modeling_ilama.IlamaModel"
+	"AutoModel": "modeling_ilama.IlamaModel",
+    "AutoModelForCausalLM": "modeling_ilama.IlamaForCausalLM"
   }
 }