Update config.json

config.json

@@ -28,5 +28,10 @@
   "summary_use_proj": true,
   "transformers_version": "4.57.3",
   "use_cache": true,
-  "vocab_size": 30000
-}
+  "vocab_size": 30000,
+  "use_safetensors": true,
+  "torch_dtype": "float32",
+  "auto_map": {
+    "AutoModelForCausalLM": "modeling_gpt2.GPT2LMHeadModel"
+  }
+}