yujiepan commited on
Commit
7a6a312
·
verified ·
1 Parent(s): 45d961d

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +34 -0
README.md CHANGED
@@ -112,6 +112,40 @@ model.save_pretrained(save_folder)
112
 
113
  </details>
114
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
115
 
116
  ### Test environment:
117
 
 
112
 
113
  </details>
114
 
115
+ ### Printing the model:
116
+
117
+ <details><summary>Click to expand</summary>
118
+
119
+ ```text
120
+ Qwen2ForCausalLM(
121
+ (model): Qwen2Model(
122
+ (embed_tokens): Embedding(152064, 8)
123
+ (layers): ModuleList(
124
+ (0-3): 4 x Qwen2DecoderLayer(
125
+ (self_attn): Qwen2Attention(
126
+ (q_proj): Linear(in_features=8, out_features=256, bias=True)
127
+ (k_proj): Linear(in_features=8, out_features=128, bias=True)
128
+ (v_proj): Linear(in_features=8, out_features=128, bias=True)
129
+ (o_proj): Linear(in_features=256, out_features=8, bias=False)
130
+ )
131
+ (mlp): Qwen2MLP(
132
+ (gate_proj): Linear(in_features=8, out_features=32, bias=False)
133
+ (up_proj): Linear(in_features=8, out_features=32, bias=False)
134
+ (down_proj): Linear(in_features=32, out_features=8, bias=False)
135
+ (act_fn): SiLUActivation()
136
+ )
137
+ (input_layernorm): Qwen2RMSNorm((8,), eps=1e-06)
138
+ (post_attention_layernorm): Qwen2RMSNorm((8,), eps=1e-06)
139
+ )
140
+ )
141
+ (norm): Qwen2RMSNorm((8,), eps=1e-06)
142
+ (rotary_emb): Qwen2RotaryEmbedding()
143
+ )
144
+ (lm_head): Linear(in_features=8, out_features=152064, bias=False)
145
+ )
146
+ ```
147
+
148
+ </details>
149
 
150
  ### Test environment:
151