config.json

{
  "hidden_size": 128,
  "model_type": "custom-model",
  "transformers_version": "4.44.2",
  "auto_map": {
    "AutoModel": "custom_model.CustomModel",
    "AutoConfig": "custom_model.CustomModelConfig"
  },
  "architectures": [
    "CustomLM"
  ]
}