baileyk's picture
Patch checkpoint to transformers main format (convert_olmo_hybrid_hf_to_main)
5bc6a76 verified
{
"architectures": [
"OlmoHybridForCausalLM"
],
"attention_bias": false,
"attention_dropout": 0.0,
"dtype": "bfloat16",
"eos_token_id": 100257,
"hidden_act": "silu",
"hidden_size": 3840,
"initializer_range": 0.02,
"intermediate_size": 11008,
"layer_types": [
"linear_attention",
"linear_attention",
"linear_attention",
"full_attention",
"linear_attention",
"linear_attention",
"linear_attention",
"full_attention",
"linear_attention",
"linear_attention",
"linear_attention",
"full_attention",
"linear_attention",
"linear_attention",
"linear_attention",
"full_attention",
"linear_attention",
"linear_attention",
"linear_attention",
"full_attention",
"linear_attention",
"linear_attention",
"linear_attention",
"full_attention",
"linear_attention",
"linear_attention",
"linear_attention",
"full_attention",
"linear_attention",
"linear_attention",
"linear_attention",
"full_attention"
],
"linear_allow_neg_eigval": true,
"linear_conv_kernel_dim": 4,
"linear_key_head_dim": 96,
"linear_num_key_heads": 30,
"linear_num_value_heads": 30,
"linear_value_head_dim": 192,
"max_position_embeddings": 32768,
"model_type": "olmo_hybrid",
"num_attention_heads": 30,
"num_hidden_layers": 32,
"num_key_value_heads": 30,
"pad_token_id": 100277,
"rms_norm_eps": 1e-06,
"rope_parameters": {
"rope_theta": null,
"rope_type": "default"
},
"tie_word_embeddings": false,
"transformers_version": "5.0.0.dev0",
"use_cache": true,
"vocab_size": 100352
}