Hugging Face's logo

shibatch
/
tinydiffusiongemmamoe4m

Text Generation
Transformers
Safetensors
English
diffusion-gemma
block-diffusion
mixture-of-experts
Mixture of Experts
tinystories
tiny-model
validation
debug-model
Model card Files
xet
 Community
tinydiffusiongemmamoe4m
File size: 3,453 Bytes
d657346
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
{
  "transformers_version": "5.14.1",
  "architectures": [
    "DiffusionGemmaForBlockDiffusion"
  ],
  "output_hidden_states": false,
  "return_dict": true,
  "dtype": "float32",
  "chunk_size_feed_forward": 0,
  "is_encoder_decoder": false,
  "id2label": {
    "0": "LABEL_0",
    "1": "LABEL_1"
  },
  "label2id": {
    "LABEL_0": 0,
    "LABEL_1": 1
  },
  "problem_type": null,
  "text_config": {
    "architectures": null,
    "output_hidden_states": false,
    "return_dict": true,
    "dtype": "float32",
    "chunk_size_feed_forward": 0,
    "is_encoder_decoder": false,
    "id2label": {
      "0": "LABEL_0",
      "1": "LABEL_1"
    },
    "label2id": {
      "LABEL_0": 0,
      "LABEL_1": 1
    },
    "problem_type": null,
    "vocab_size": 1003,
    "hidden_size": 128,
    "intermediate_size": 384,
    "num_hidden_layers": 8,
    "num_attention_heads": 4,
    "num_key_value_heads": 1,
    "head_dim": 32,
    "hidden_activation": "gelu_pytorch_tanh",
    "max_position_embeddings": 1024,
    "initializer_range": 0.02,
    "rms_norm_eps": 1e-06,
    "pad_token_id": 1000,
    "eos_token_id": 1001,
    "bos_token_id": 1000,
    "tie_word_embeddings": true,
    "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"
      }
    },
    "attention_bias": false,
    "attention_dropout": 0.0,
    "sliding_window": 64,
    "layer_types": [
      "sliding_attention",
      "sliding_attention",
      "sliding_attention",
      "sliding_attention",
      "sliding_attention",
      "full_attention",
      "sliding_attention",
      "full_attention"
    ],
    "use_bidirectional_attention": "vision",
    "num_global_key_value_heads": 1,
    "global_head_dim": 32,
    "num_experts": 4,
    "top_k_experts": 2,
    "moe_intermediate_size": 192,
    "_name_or_path": "",
    "model_type": "diffusion_gemma_text",
    "output_attentions": false
  },
  "vision_config": {
    "architectures": null,
    "output_hidden_states": false,
    "return_dict": true,
    "dtype": "float32",
    "chunk_size_feed_forward": 0,
    "is_encoder_decoder": false,
    "id2label": {
      "0": "LABEL_0",
      "1": "LABEL_1"
    },
    "label2id": {
      "LABEL_0": 0,
      "LABEL_1": 1
    },
    "problem_type": null,
    "hidden_size": 32,
    "intermediate_size": 64,
    "num_hidden_layers": 1,
    "num_attention_heads": 4,
    "num_key_value_heads": 4,
    "head_dim": 8,
    "hidden_activation": "gelu_pytorch_tanh",
    "rms_norm_eps": 1e-06,
    "max_position_embeddings": 256,
    "attention_bias": false,
    "attention_dropout": 0.0,
    "rope_parameters": {
      "rope_theta": 100.0,
      "rope_type": "default"
    },
    "pooling_kernel_size": 1,
    "patch_size": 16,
    "position_embedding_size": 64,
    "use_clipped_linears": false,
    "standardize": false,
    "initializer_range": 0.02,
    "_name_or_path": "",
    "model_type": "gemma4_vision",
    "output_attentions": false
  },
  "boi_token_id": 255999,
  "eoi_token_id": 258882,
  "image_token_id": 258880,
  "initializer_range": 0.02,
  "tie_word_embeddings": true,
  "canvas_length": 8,
  "_name_or_path": "/home/codex/tinyModels/runs/diffusion_gemma_100p_scratch_full220k/checkpoints/full-0080000/hf",
  "model_type": "diffusion_gemma",
  "output_attentions": false
}