google/diffusiongemma-26B-A4B-it
✓ Same architecture, smaller
Every module class, layer kind and checked property of the original is in the fixture.
Model tester (advisory)
The model's own small test config in its ModelTester (10 literal values, 10 set by the model's own tester, the rest generic base-tester defaults). Testers are usually representative but not always, so these notes are for review, not rules.
The fixture covers every per-layer kind the tester sets, and shrinks every size it does.
Layer stacks
One square per layer, colored by kind: the layer's module classes and type attributes. Hover a square for its index and kind.
model.encoder.language_model.layers
| Kind | Original | Fixture | |
|---|---|---|---|
| sliding_attention | 25 | 1 | ✓ kept |
| full_attention | 5 | 1 | ✓ kept |
model.encoder.vision_tower.encoder.layers
| Kind | Original | Fixture | |
|---|---|---|---|
| Gemma4VisionEncoderLayer | 27 | 2 | ✓ kept |
model.decoder.layers
| Kind | Original | Fixture | |
|---|---|---|---|
| sliding_attention | 25 | 1 | ✓ kept |
| full_attention | 5 | 1 | ✓ kept |
Where the parameters are
Share of each model's parameters per component. Vocabulary and patch sizes are kept, so embeddings usually take a larger share of the fixture.
| Component | Original | Share | Fixture | Share |
|---|---|---|---|---|
| model.encoder.language_model.layers | 24.5B | 94.9% | 440.9K | 2.4% |
| model.encoder | 765.9M | 3.0% | 18.1M | 97.1% |
| model.encoder.vision_tower.encoder.layers | 545.1M | 2.1% | 82.5K | 0.4% |
| model.decoder | 17.8M | 0.1% | 24.6K | 0.1% |
Preserved properties
| Config | Property | Original | Fixture | |
|---|---|---|---|---|
| model | model_type | diffusion_gemma | diffusion_gemma | ✓ kept |
| model | tie_word_embeddings | True | True | ✓ kept |
| text_config | attention style | GQA (16 heads, 8 KV) | GQA (4 heads, 2 KV) | ✓ kept |
| text_config | model_type | diffusion_gemma_text | diffusion_gemma_text | ✓ kept |
| text_config | vocab_size | 262144 | 262144 | ✓ kept |
| text_config | tie_word_embeddings | True | True | ✓ kept |
| text_config | max_position_embeddings | 262144 | 262144 | ✓ kept |
| vision_config | attention style | MHA (16 heads, 16 KV) | MHA (4 heads, 4 KV) | ✓ kept |
| vision_config | RoPE type | axial | axial | ✓ kept |
| vision_config | model_type | gemma4_vision | gemma4_vision | ✓ kept |
| vision_config | max_position_embeddings | 131072 | 131072 | ✓ kept |
| vision_config | patch_size | 16 | 16 | ✓ kept |
Module classes
Instances of each class in the built model (prefix DiffusionGemma omitted).
| Class | Original | Fixture |
|---|---|---|
| RMSNorm | 665 | 49 |
| Linear | 665 | 50 |
| Gemma4ClippableLinear | 189 | 14 |
| Gemma4RMSNorm | 189 | 14 |
| GELUTanh | 148 | 11 |
| Text4MLP | 60 | 4 |
| TextExperts | 60 | 4 |
| TextRouter | 60 | 4 |
| DecoderTextAttention | 30 | 2 |
| DecoderTextLayer | 30 | 2 |
| EncoderTextAttention | 30 | 2 |
| EncoderTextLayer | 30 | 2 |
| Gemma4VisionAttention | 27 | 2 |
| Gemma4VisionEncoderLayer | 27 | 2 |
| Gemma4VisionMLP | 27 | 2 |
| ModuleList | 3 | 3 |
| TextRotaryEmbedding | 2 | 2 |
| TextScaledWordEmbedding | 2 | 2 |
| DecoderModel | 1 | 1 |
| EncoderModel | 1 | 1 |
| EncoderTextModel | 1 | 1 |
| ForBlockDiffusion | 1 | 1 |
| Model | 1 | 1 |
| MultimodalEmbedder | 1 | 1 |
| SelfConditioning | 1 | 1 |
| Gemma4VisionEncoder | 1 | 1 |
| Gemma4VisionModel | 1 | 1 |
| Gemma4VisionPatchEmbedder | 1 | 1 |
| Gemma4VisionPooler | 1 | 1 |
| Gemma4VisionRotaryEmbedding | 1 | 1 |
Config changes
22 fields changed, 64 unchanged.
| Field | Original | Fixture |
|---|---|---|
| text_config.head_dim | 256 | 16 |
| text_config.hidden_size | 2816 | 64 |
| text_config.intermediate_size | 2112 | 128 |
| text_config.layer_types | 30 entries: 25× sliding_attention, 5× full_attention | ["sliding_attention", "full_attention"] |
| text_config.moe_intermediate_size | 704 | 32 |
| text_config.num_attention_heads | 16 | 4 |
| text_config.num_experts | 128 | 4 |
| text_config.num_hidden_layers | 30 | 2 |
| text_config.num_key_value_heads | 8 | 2 |
| text_config.per_layer_config.05 | {"head_dim": 512, "num_key_value_heads": 2} | — |
| text_config.per_layer_config.1 | — | {"head_dim": 512} |
| text_config.per_layer_config.11 | {"head_dim": 512, "num_key_value_heads": 2} | — |
| text_config.per_layer_config.17 | {"head_dim": 512, "num_key_value_heads": 2} | — |
| text_config.per_layer_config.23 | {"head_dim": 512, "num_key_value_heads": 2} | — |
| text_config.per_layer_config.29 | {"head_dim": 512, "num_key_value_heads": 2} | — |
| text_config.top_k_experts | 8 | 2 |
| vision_config.head_dim | 72 | 16 |
| vision_config.hidden_size | 1152 | 64 |
| vision_config.intermediate_size | 4304 | 128 |
| vision_config.num_attention_heads | 16 | 4 |
| vision_config.num_hidden_layers | 27 | 2 |
| vision_config.num_key_value_heads | 16 | 4 |