rahul7star/

rahul7star-gemma-merged

Most layers: head_dim = 256
Full-attention layers: head_dim = 512
35 layers total
Full-attention layers: 4, 9, 14, 19, 24, 29, 34

from transformers import AutoConfig

cfg = AutoConfig.from_pretrained(
    "rahul7star/rahul7star-gemma-merged",
    trust_remote_code=True
)

print(type(cfg.text_config))
print(cfg.text_config)

print("num_hidden_layers =", getattr(cfg.text_config, "num_hidden_layers", None))
print("head_dim =", getattr(cfg.text_config, "head_dim", None))
print("hidden_size =", getattr(cfg.text_config, "hidden_size", None))
print("num_attention_heads =", getattr(cfg.text_config, "num_attention_heads", None))

<class 'transformers.models.gemma4.configuration_gemma4.Gemma4TextConfig'>
Gemma4TextConfig {
  "attention_bias": false,
  "attention_dropout": 0.0,
  "attention_k_eq_v": false,
  "bos_token_id": 2,
  "dtype": "bfloat16",
  "enable_moe_block": false,
  "eos_token_id": 1,
  "expert_intermediate_size": null,
  "final_logit_softcapping": 30.0,
  "head_dim": 256,
  "hidden_activation": "gelu_pytorch_tanh",
  "hidden_size": 1536,
  "hidden_size_per_layer_input": 256,
  "initializer_range": 0.02,
  "intermediate_size": 6144,
  "layer_types": [
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "full_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "full_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "full_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "full_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "full_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "full_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "full_attention"
  ],
  "max_position_embeddings": 131072,
  "model_type": "gemma4_text",
  "moe_intermediate_size": null,
  "num_attention_heads": 8,
  "num_experts": null,
  "num_hidden_layers": 35,
  "num_key_value_heads": 1,
  "num_kv_shared_layers": 20,
  "pad_token_id": 0,
  "per_layer_config": {
    "04": {
      "head_dim": 512
    },
    "09": {
      "head_dim": 512
    },
    "14": {
      "head_dim": 512
    },
    "19": {
      "head_dim": 512
    },
    "24": {
      "head_dim": 512
    },
    "29": {
      "head_dim": 512
    },
    "34": {
      "head_dim": 512
    }
  },
  "rms_norm_eps": 1e-06,
  "rope_parameters": {
    "full_attention": {
      "partial_rotary_factor": 0.25,
      "rope_theta": 1000000.0,
      "rope_type": "proportional"
    },
    "sliding_attention": {
      "rope_theta": 10000.0,
      "rope_type": "default"
    }
  },
  "sliding_window": 512,
  "tie_word_embeddings": true,
  "top_k_experts": null,
  "transformers_version": "5.15.0",
  "use_bidirectional_attention": null,
  "use_cache": true,
  "use_double_wide_mlp": true,
  "vocab_size": 262144,
  "vocab_size_per_layer_input": 262144
}

num_hidden_layers = 35