ereniko's picture
Add safetensors + Transformers (AutoModelForCausalLM) support
5974fd0 verified
Raw
History Blame Contribute Delete
592 Bytes
{
"architectures": [
"IvmeForCausalLM"
],
"auto_map": {
"AutoConfig": "configuration_ivme.IvmeConfig",
"AutoModelForCausalLM": "modeling_ivme.IvmeForCausalLM"
},
"context_len": 1024,
"dropout": 0.0,
"dtype": "float32",
"ffn_mult": 4.0,
"hidden_dim": 384,
"hidden_size": 384,
"max_position_embeddings": 1024,
"model_type": "ivme",
"n_heads": 6,
"n_layers": 10,
"norm_eps": 1e-05,
"num_attention_heads": 6,
"num_hidden_layers": 10,
"rope_theta": 10000.0,
"tie_word_embeddings": true,
"transformers_version": "5.13.1",
"vocab_size": 16000
}