config.json

{
  "architectures": [
    "FrozenLMModel"
  ],
  "auto_map": {
    "AutoConfig": "modeling_frozen_lm_mlp.FrozenLMConfig",
    "AutoModel": "modeling_frozen_lm_mlp.FrozenLMModel"
  },
  "hidden_dim": 128,
  "input_dim": 256,
  "output_dim": 24576,
  "model_type": "frozen_lm_mlp",
  "torch_dtype": "float32",
  "transformers_version": "4.36.0"
}