Enigma / config.json
dr-tkxx's picture
Upload 8 files
2258d52 verified
{
"architectures": [
"EnigmaForCausalLM"
],
"auto_map": {
"AutoConfig": "enigma_module.EnigmaConfig",
"AutoModel": "enigma_module.EnigmaModel",
"AutoModelForCausalLM": "enigma_module.EnigmaForCausalLM"
},
"dtype": "float32",
"hidden_size": 128,
"is_decoder": true,
"model_type": "enigma",
"num_attention_heads": 1,
"num_hidden_layers": 1,
"transformers_version": "5.8.0.dev0",
"vocab_size": 50257
}