File size: 62,485 Bytes
7d5fc54 | 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 | Valid precision + execution provider combinations are: FP32 CPU, FP32 CUDA, FP16 CUDA, FP16 DML, FP16 TRT-RTX, BF16 CUDA, BF16 TRT-RTX, INT8 CPU, INT8 CUDA, INT8 WebGPU, INT4 CPU, INT4 CUDA, INT4 DML, INT4 WebGPU
Extra options: {}
GroupQueryAttention (GQA) is used in this model.
GroupQueryAttention (GQA) is used in this model.
Loading weights: 0%| | 0/426 [00:00<?, ?it/s]
Loading weights: 16%|ββ | 69/426 [00:00<00:00, 672.81it/s]
Loading weights: 34%|ββββ | 144/426 [00:00<00:00, 715.00it/s]
Loading weights: 52%|ββββββ | 221/426 [00:00<00:00, 735.44it/s]
Loading weights: 69%|βββββββ | 296/426 [00:00<00:00, 719.97it/s]
Loading weights: 89%|βββββββββ | 379/426 [00:00<00:00, 754.67it/s]
Loading weights: 100%|ββββββββββ| 426/426 [00:00<00:00, 754.59it/s]
Reading embedding layer
Reading layer 0
Reading layer 1
Reading layer 2
Reading layer 3
Reading layer 4
Reading layer 5
Reading layer 6
Reading layer 7
Reading layer 8
Reading layer 9
Reading layer 10
Reading layer 11
Reading layer 12
Reading layer 13
Reading layer 14
Reading layer 15
Reading layer 16
Reading layer 17
Reading layer 18
Reading layer 19
Reading layer 20
Reading layer 21
Reading layer 22
Reading layer 23
Reading layer 24
Reading layer 25
Reading layer 26
Reading layer 27
Reading layer 28
Reading layer 29
Reading layer 30
Reading layer 31
Reading final norm
Reading LM head
Building MTP (multi-token prediction) head -> mtp.onnx
Saving ONNX model in /home/anon/knowline/onnx/model
0it [00:00, ?it/s]
Saving model.layers.0.input_layernorm.weight (f16, [2560]): 0%| | 1/404 [00:00<00:01, 257.64it/s]
Saving model.layers.0.linear_attn.qkv_proj.MatMul.weight (f16, [2560,8192]): 0%| | 2/404 [00:00<00:00, 479.76it/s]
Saving model.layers.0.linear_attn.z_proj.MatMul.weight (f16, [2560,4096]): 1%| | 3/404 [00:00<00:03, 102.33it/s]
Saving model.layers.0.linear_attn.b_proj.MatMul.weight (f16, [2560,32]): 1%| | 4/404 [00:00<00:04, 92.50it/s]
Saving model.layers.0.linear_attn.a_proj.MatMul.weight (f16, [2560,32]): 1%| | 5/404 [00:00<00:03, 114.79it/s]
Saving model.layers.0.linear_attn.conv1d.weight (f16, [8192,1,4]): 1%|β | 6/404 [00:00<00:02, 137.16it/s]
Saving model.layers.0.linear_attn.conv1d.bias (f16, [8192]): 2%|β | 7/404 [00:00<00:02, 159.58it/s]
Saving model.layers.0.linear_attn.A_log (f32, [32]): 2%|β | 8/404 [00:00<00:02, 181.96it/s]
Saving model.layers.0.linear_attn.dt_bias (f32, [32]): 2%|β | 9/404 [00:00<00:01, 204.30it/s]
Saving model.layers.0.linear_attn.norm.weight (f16, [128]): 2%|β | 10/404 [00:00<00:01, 226.72it/s]
Saving model.layers.0.linear_attn.out_proj.MatMul.weight (f16, [4096,2560]): 3%|β | 11/404 [00:00<00:01, 249.07it/s]
Saving model.layers.0.post_attention_layernorm.weight (f16, [2560]): 3%|β | 12/404 [00:00<00:01, 208.07it/s]
Saving model.layers.0.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 3%|β | 13/404 [00:00<00:01, 224.36it/s]
Saving model.layers.0.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 3%|β | 14/404 [00:00<00:13, 29.58it/s]
Saving model.layers.0.mlp.down_proj.MatMul.weight (f16, [9216,2560]): 3%|β | 14/404 [00:00<00:13, 29.58it/s]
Saving model.layers.1.input_layernorm.weight (f16, [2560]): 4%|β | 15/404 [00:00<00:13, 29.58it/s]
Saving model.layers.1.linear_attn.qkv_proj.MatMul.weight (f16, [2560,8192]): 4%|β | 16/404 [00:00<00:13, 29.58it/s]
Saving model.layers.1.linear_attn.z_proj.MatMul.weight (f16, [2560,4096]): 4%|β | 17/404 [00:00<00:13, 29.58it/s]
Saving model.layers.1.linear_attn.b_proj.MatMul.weight (f16, [2560,32]): 4%|β | 18/404 [00:00<00:13, 29.58it/s]
Saving model.layers.1.linear_attn.a_proj.MatMul.weight (f16, [2560,32]): 5%|β | 19/404 [00:00<00:13, 29.58it/s]
Saving model.layers.1.linear_attn.conv1d.weight (f16, [8192,1,4]): 5%|β | 20/404 [00:00<00:12, 29.58it/s]
Saving model.layers.1.linear_attn.conv1d.bias (f16, [8192]): 5%|β | 21/404 [00:00<00:12, 29.58it/s]
Saving model.layers.1.linear_attn.A_log (f32, [32]): 5%|β | 22/404 [00:00<00:12, 29.58it/s]
Saving model.layers.1.linear_attn.dt_bias (f32, [32]): 6%|β | 23/404 [00:00<00:12, 29.58it/s]
Saving model.layers.1.linear_attn.norm.weight (f16, [128]): 6%|β | 24/404 [00:00<00:12, 29.58it/s]
Saving model.layers.1.linear_attn.out_proj.MatMul.weight (f16, [4096,2560]): 6%|β | 25/404 [00:00<00:12, 29.58it/s]
Saving model.layers.1.post_attention_layernorm.weight (f16, [2560]): 6%|β | 26/404 [00:00<00:12, 29.58it/s]
Saving model.layers.1.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 7%|β | 27/404 [00:00<00:12, 29.58it/s]
Saving model.layers.1.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 7%|β | 28/404 [00:00<00:13, 28.42it/s]
Saving model.layers.1.mlp.down_proj.MatMul.weight (f16, [9216,2560]): 7%|β | 28/404 [00:00<00:13, 28.42it/s]
Saving model.layers.2.input_layernorm.weight (f16, [2560]): 7%|β | 29/404 [00:01<00:13, 28.42it/s]
Saving model.layers.2.linear_attn.qkv_proj.MatMul.weight (f16, [2560,8192]): 7%|β | 30/404 [00:01<00:13, 28.42it/s]
Saving model.layers.2.linear_attn.z_proj.MatMul.weight (f16, [2560,4096]): 8%|β | 31/404 [00:01<00:13, 28.42it/s]
Saving model.layers.2.linear_attn.b_proj.MatMul.weight (f16, [2560,32]): 8%|β | 32/404 [00:01<00:13, 28.42it/s]
Saving model.layers.2.linear_attn.a_proj.MatMul.weight (f16, [2560,32]): 8%|β | 33/404 [00:01<00:13, 28.42it/s]
Saving model.layers.2.linear_attn.conv1d.weight (f16, [8192,1,4]): 8%|β | 34/404 [00:01<00:13, 28.42it/s]
Saving model.layers.2.linear_attn.conv1d.bias (f16, [8192]): 9%|β | 35/404 [00:01<00:12, 28.42it/s]
Saving model.layers.2.linear_attn.A_log (f32, [32]): 9%|β | 36/404 [00:01<00:12, 28.42it/s]
Saving model.layers.2.linear_attn.dt_bias (f32, [32]): 9%|β | 37/404 [00:01<00:12, 28.42it/s]
Saving model.layers.2.linear_attn.norm.weight (f16, [128]): 9%|β | 38/404 [00:01<00:12, 28.42it/s]
Saving model.layers.2.linear_attn.out_proj.MatMul.weight (f16, [4096,2560]): 10%|β | 39/404 [00:01<00:12, 28.42it/s]
Saving model.layers.2.post_attention_layernorm.weight (f16, [2560]): 10%|β | 40/404 [00:01<00:12, 28.42it/s]
Saving model.layers.2.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 10%|β | 41/404 [00:01<00:12, 28.42it/s]
Saving model.layers.2.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 10%|β | 42/404 [00:01<00:12, 28.04it/s]
Saving model.layers.2.mlp.down_proj.MatMul.weight (f16, [9216,2560]): 10%|β | 42/404 [00:01<00:12, 28.04it/s]
Saving model.layers.3.input_layernorm.weight (f16, [2560]): 11%|β | 43/404 [00:01<00:12, 28.04it/s]
Saving model.layers.3.attn.qkv_proj.MatMul.weight (f16, [2560,10240]): 11%|β | 44/404 [00:01<00:12, 28.04it/s]
Saving model.layers.3.attn.q_norm.layernorm.weight (f16, [256]): 11%|β | 45/404 [00:01<00:12, 28.04it/s]
Saving model.layers.3.attn.k_norm.layernorm.weight (f16, [256]): 11%|ββ | 46/404 [00:01<00:12, 28.04it/s]
Saving cos_cache (f32, [262144,32]): 12%|ββ | 47/404 [00:01<00:12, 28.04it/s]
Saving sin_cache (f32, [262144,32]): 12%|ββ | 48/404 [00:01<00:12, 28.04it/s]
Saving model.layers.3.attn.o_proj.MatMul.weight (f16, [4096,2560]): 12%|ββ | 49/404 [00:01<00:12, 28.04it/s]
Saving model.layers.3.post_attention_layernorm.weight (f16, [2560]): 12%|ββ | 50/404 [00:01<00:12, 28.04it/s]
Saving model.layers.3.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 13%|ββ | 51/404 [00:01<00:12, 28.04it/s]
Saving model.layers.3.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 13%|ββ | 52/404 [00:01<00:14, 24.68it/s]
Saving model.layers.3.mlp.down_proj.MatMul.weight (f16, [9216,2560]): 13%|ββ | 52/404 [00:01<00:14, 24.68it/s]
Saving model.layers.4.input_layernorm.weight (f16, [2560]): 13%|ββ | 53/404 [00:02<00:14, 24.68it/s]
Saving model.layers.4.linear_attn.qkv_proj.MatMul.weight (f16, [2560,8192]): 13%|ββ | 54/404 [00:02<00:14, 24.68it/s]
Saving model.layers.4.linear_attn.z_proj.MatMul.weight (f16, [2560,4096]): 14%|ββ | 55/404 [00:02<00:14, 24.68it/s]
Saving model.layers.4.linear_attn.b_proj.MatMul.weight (f16, [2560,32]): 14%|ββ | 56/404 [00:02<00:14, 24.68it/s]
Saving model.layers.4.linear_attn.a_proj.MatMul.weight (f16, [2560,32]): 14%|ββ | 57/404 [00:02<00:14, 24.68it/s]
Saving model.layers.4.linear_attn.conv1d.weight (f16, [8192,1,4]): 14%|ββ | 58/404 [00:02<00:14, 24.68it/s]
Saving model.layers.4.linear_attn.conv1d.bias (f16, [8192]): 15%|ββ | 59/404 [00:02<00:13, 24.68it/s]
Saving model.layers.4.linear_attn.A_log (f32, [32]): 15%|ββ | 60/404 [00:02<00:13, 24.68it/s]
Saving model.layers.4.linear_attn.dt_bias (f32, [32]): 15%|ββ | 61/404 [00:02<00:13, 24.68it/s]
Saving model.layers.4.linear_attn.norm.weight (f16, [128]): 15%|ββ | 62/404 [00:02<00:13, 24.68it/s]
Saving model.layers.4.linear_attn.out_proj.MatMul.weight (f16, [4096,2560]): 16%|ββ | 63/404 [00:02<00:13, 24.68it/s]
Saving model.layers.4.linear_attn.out_proj.MatMul.weight (f16, [4096,2560]): 16%|ββ | 64/404 [00:02<00:09, 34.15it/s]
Saving model.layers.4.post_attention_layernorm.weight (f16, [2560]): 16%|ββ | 64/404 [00:02<00:09, 34.15it/s]
Saving model.layers.4.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 16%|ββ | 65/404 [00:02<00:09, 34.15it/s]
Saving model.layers.4.mlp.down_proj.MatMul.weight (f16, [9216,2560]): 16%|ββ | 66/404 [00:02<00:09, 34.15it/s]
Saving model.layers.5.input_layernorm.weight (f16, [2560]): 17%|ββ | 67/404 [00:02<00:09, 34.15it/s]
Saving model.layers.5.linear_attn.qkv_proj.MatMul.weight (f16, [2560,8192]): 17%|ββ | 68/404 [00:02<00:09, 34.15it/s]
Saving model.layers.5.linear_attn.z_proj.MatMul.weight (f16, [2560,4096]): 17%|ββ | 69/404 [00:02<00:09, 34.15it/s]
Saving model.layers.5.linear_attn.z_proj.MatMul.weight (f16, [2560,4096]): 17%|ββ | 70/404 [00:02<00:13, 25.32it/s]
Saving model.layers.5.linear_attn.b_proj.MatMul.weight (f16, [2560,32]): 17%|ββ | 70/404 [00:02<00:13, 25.32it/s]
Saving model.layers.5.linear_attn.a_proj.MatMul.weight (f16, [2560,32]): 18%|ββ | 71/404 [00:02<00:13, 25.32it/s]
Saving model.layers.5.linear_attn.conv1d.weight (f16, [8192,1,4]): 18%|ββ | 72/404 [00:02<00:13, 25.32it/s]
Saving model.layers.5.linear_attn.conv1d.bias (f16, [8192]): 18%|ββ | 73/404 [00:02<00:13, 25.32it/s]
Saving model.layers.5.linear_attn.A_log (f32, [32]): 18%|ββ | 74/404 [00:02<00:13, 25.32it/s]
Saving model.layers.5.linear_attn.dt_bias (f32, [32]): 19%|ββ | 75/404 [00:02<00:12, 25.32it/s]
Saving model.layers.5.linear_attn.norm.weight (f16, [128]): 19%|ββ | 76/404 [00:02<00:12, 25.32it/s]
Saving model.layers.5.linear_attn.out_proj.MatMul.weight (f16, [4096,2560]): 19%|ββ | 77/404 [00:02<00:12, 25.32it/s]
Saving model.layers.5.post_attention_layernorm.weight (f16, [2560]): 19%|ββ | 78/404 [00:02<00:12, 25.32it/s]
Saving model.layers.5.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 20%|ββ | 79/404 [00:02<00:12, 25.32it/s]
Saving model.layers.5.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 20%|ββ | 80/404 [00:03<00:13, 24.43it/s]
Saving model.layers.5.mlp.down_proj.MatMul.weight (f16, [9216,2560]): 20%|ββ | 80/404 [00:03<00:13, 24.43it/s]
Saving model.layers.6.input_layernorm.weight (f16, [2560]): 20%|ββ | 81/404 [00:03<00:13, 24.43it/s]
Saving model.layers.6.linear_attn.qkv_proj.MatMul.weight (f16, [2560,8192]): 20%|ββ | 82/404 [00:03<00:13, 24.43it/s]
Saving model.layers.6.linear_attn.z_proj.MatMul.weight (f16, [2560,4096]): 21%|ββ | 83/404 [00:03<00:13, 24.43it/s]
Saving model.layers.6.linear_attn.b_proj.MatMul.weight (f16, [2560,32]): 21%|ββ | 84/404 [00:03<00:13, 24.43it/s]
Saving model.layers.6.linear_attn.a_proj.MatMul.weight (f16, [2560,32]): 21%|ββ | 85/404 [00:03<00:13, 24.43it/s]
Saving model.layers.6.linear_attn.conv1d.weight (f16, [8192,1,4]): 21%|βββ | 86/404 [00:03<00:13, 24.43it/s]
Saving model.layers.6.linear_attn.conv1d.bias (f16, [8192]): 22%|βββ | 87/404 [00:03<00:12, 24.43it/s]
Saving model.layers.6.linear_attn.A_log (f32, [32]): 22%|βββ | 88/404 [00:03<00:12, 24.43it/s]
Saving model.layers.6.linear_attn.dt_bias (f32, [32]): 22%|βββ | 89/404 [00:03<00:12, 24.43it/s]
Saving model.layers.6.linear_attn.norm.weight (f16, [128]): 22%|βββ | 90/404 [00:03<00:12, 24.43it/s]
Saving model.layers.6.linear_attn.out_proj.MatMul.weight (f16, [4096,2560]): 23%|βββ | 91/404 [00:03<00:12, 24.43it/s]
Saving model.layers.6.post_attention_layernorm.weight (f16, [2560]): 23%|βββ | 92/404 [00:03<00:12, 24.43it/s]
Saving model.layers.6.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 23%|βββ | 93/404 [00:03<00:12, 24.43it/s]
Saving model.layers.6.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 23%|βββ | 94/404 [00:03<00:12, 25.59it/s]
Saving model.layers.6.mlp.down_proj.MatMul.weight (f16, [9216,2560]): 23%|βββ | 94/404 [00:03<00:12, 25.59it/s]
Saving model.layers.7.input_layernorm.weight (f16, [2560]): 24%|βββ | 95/404 [00:03<00:12, 25.59it/s]
Saving model.layers.7.attn.qkv_proj.MatMul.weight (f16, [2560,10240]): 24%|βββ | 96/404 [00:03<00:12, 25.59it/s]
Saving model.layers.7.attn.q_norm.layernorm.weight (f16, [256]): 24%|βββ | 97/404 [00:03<00:11, 25.59it/s]
Saving model.layers.7.attn.k_norm.layernorm.weight (f16, [256]): 24%|βββ | 98/404 [00:03<00:11, 25.59it/s]
Saving model.layers.7.attn.o_proj.MatMul.weight (f16, [4096,2560]): 25%|βββ | 99/404 [00:03<00:11, 25.59it/s]
Saving model.layers.7.post_attention_layernorm.weight (f16, [2560]): 25%|βββ | 100/404 [00:03<00:11, 25.59it/s]
Saving model.layers.7.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 25%|βββ | 101/404 [00:03<00:11, 25.59it/s]
Saving model.layers.7.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 25%|βββ | 102/404 [00:04<00:13, 22.30it/s]
Saving model.layers.7.mlp.down_proj.MatMul.weight (f16, [9216,2560]): 25%|βββ | 102/404 [00:04<00:13, 22.30it/s]
Saving model.layers.8.input_layernorm.weight (f16, [2560]): 25%|βββ | 103/404 [00:04<00:13, 22.30it/s]
Saving model.layers.8.linear_attn.qkv_proj.MatMul.weight (f16, [2560,8192]): 26%|βββ | 104/404 [00:04<00:13, 22.30it/s]
Saving model.layers.8.linear_attn.z_proj.MatMul.weight (f16, [2560,4096]): 26%|βββ | 105/404 [00:04<00:13, 22.30it/s]
Saving model.layers.8.linear_attn.b_proj.MatMul.weight (f16, [2560,32]): 26%|βββ | 106/404 [00:04<00:13, 22.30it/s]
Saving model.layers.8.linear_attn.a_proj.MatMul.weight (f16, [2560,32]): 26%|βββ | 107/404 [00:04<00:13, 22.30it/s]
Saving model.layers.8.linear_attn.conv1d.weight (f16, [8192,1,4]): 27%|βββ | 108/404 [00:04<00:13, 22.30it/s]
Saving model.layers.8.linear_attn.conv1d.bias (f16, [8192]): 27%|βββ | 109/404 [00:04<00:13, 22.30it/s]
Saving model.layers.8.linear_attn.A_log (f32, [32]): 27%|βββ | 110/404 [00:04<00:13, 22.30it/s]
Saving model.layers.8.linear_attn.dt_bias (f32, [32]): 27%|βββ | 111/404 [00:04<00:13, 22.30it/s]
Saving model.layers.8.linear_attn.norm.weight (f16, [128]): 28%|βββ | 112/404 [00:04<00:13, 22.30it/s]
Saving model.layers.8.linear_attn.out_proj.MatMul.weight (f16, [4096,2560]): 28%|βββ | 113/404 [00:04<00:13, 22.30it/s]
Saving model.layers.8.post_attention_layernorm.weight (f16, [2560]): 28%|βββ | 114/404 [00:04<00:13, 22.30it/s]
Saving model.layers.8.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 28%|βββ | 115/404 [00:04<00:12, 22.30it/s]
Saving model.layers.8.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 29%|βββ | 116/404 [00:04<00:11, 24.11it/s]
Saving model.layers.8.mlp.down_proj.MatMul.weight (f16, [9216,2560]): 29%|βββ | 116/404 [00:04<00:11, 24.11it/s]
Saving model.layers.9.input_layernorm.weight (f16, [2560]): 29%|βββ | 117/404 [00:04<00:11, 24.11it/s]
Saving model.layers.9.linear_attn.qkv_proj.MatMul.weight (f16, [2560,8192]): 29%|βββ | 118/404 [00:04<00:11, 24.11it/s]
Saving model.layers.9.linear_attn.z_proj.MatMul.weight (f16, [2560,4096]): 29%|βββ | 119/404 [00:04<00:11, 24.11it/s]
Saving model.layers.9.linear_attn.b_proj.MatMul.weight (f16, [2560,32]): 30%|βββ | 120/404 [00:04<00:11, 24.11it/s]
Saving model.layers.9.linear_attn.a_proj.MatMul.weight (f16, [2560,32]): 30%|βββ | 121/404 [00:04<00:11, 24.11it/s]
Saving model.layers.9.linear_attn.conv1d.weight (f16, [8192,1,4]): 30%|βββ | 122/404 [00:04<00:11, 24.11it/s]
Saving model.layers.9.linear_attn.conv1d.bias (f16, [8192]): 30%|βββ | 123/404 [00:04<00:11, 24.11it/s]
Saving model.layers.9.linear_attn.A_log (f32, [32]): 31%|βββ | 124/404 [00:04<00:11, 24.11it/s]
Saving model.layers.9.linear_attn.dt_bias (f32, [32]): 31%|βββ | 125/404 [00:04<00:11, 24.11it/s]
Saving model.layers.9.linear_attn.norm.weight (f16, [128]): 31%|βββ | 126/404 [00:04<00:11, 24.11it/s]
Saving model.layers.9.linear_attn.out_proj.MatMul.weight (f16, [4096,2560]): 31%|ββββ | 127/404 [00:04<00:11, 24.11it/s]
Saving model.layers.9.post_attention_layernorm.weight (f16, [2560]): 32%|ββββ | 128/404 [00:04<00:11, 24.11it/s]
Saving model.layers.9.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 32%|ββββ | 129/404 [00:04<00:11, 24.11it/s]
Saving model.layers.9.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 32%|ββββ | 130/404 [00:05<00:10, 25.22it/s]
Saving model.layers.9.mlp.down_proj.MatMul.weight (f16, [9216,2560]): 32%|ββββ | 130/404 [00:05<00:10, 25.22it/s]
Saving model.layers.10.input_layernorm.weight (f16, [2560]): 32%|ββββ | 131/404 [00:05<00:10, 25.22it/s]
Saving model.layers.10.linear_attn.qkv_proj.MatMul.weight (f16, [2560,8192]): 33%|ββββ | 132/404 [00:05<00:10, 25.22it/s]
Saving model.layers.10.linear_attn.z_proj.MatMul.weight (f16, [2560,4096]): 33%|ββββ | 133/404 [00:05<00:10, 25.22it/s]
Saving model.layers.10.linear_attn.b_proj.MatMul.weight (f16, [2560,32]): 33%|ββββ | 134/404 [00:05<00:10, 25.22it/s]
Saving model.layers.10.linear_attn.a_proj.MatMul.weight (f16, [2560,32]): 33%|ββββ | 135/404 [00:05<00:10, 25.22it/s]
Saving model.layers.10.linear_attn.conv1d.weight (f16, [8192,1,4]): 34%|ββββ | 136/404 [00:05<00:10, 25.22it/s]
Saving model.layers.10.linear_attn.conv1d.bias (f16, [8192]): 34%|ββββ | 137/404 [00:05<00:10, 25.22it/s]
Saving model.layers.10.linear_attn.A_log (f32, [32]): 34%|ββββ | 138/404 [00:05<00:10, 25.22it/s]
Saving model.layers.10.linear_attn.dt_bias (f32, [32]): 34%|ββββ | 139/404 [00:05<00:10, 25.22it/s]
Saving model.layers.10.linear_attn.norm.weight (f16, [128]): 35%|ββββ | 140/404 [00:05<00:10, 25.22it/s]
Saving model.layers.10.linear_attn.out_proj.MatMul.weight (f16, [4096,2560]): 35%|ββββ | 141/404 [00:05<00:10, 25.22it/s]
Saving model.layers.10.linear_attn.out_proj.MatMul.weight (f16, [4096,2560]): 35%|ββββ | 142/404 [00:05<00:07, 33.03it/s]
Saving model.layers.10.post_attention_layernorm.weight (f16, [2560]): 35%|ββββ | 142/404 [00:05<00:07, 33.03it/s]
Saving model.layers.10.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 35%|ββββ | 143/404 [00:05<00:07, 33.03it/s]
Saving model.layers.10.mlp.down_proj.MatMul.weight (f16, [9216,2560]): 36%|ββββ | 144/404 [00:05<00:07, 33.03it/s]
Saving model.layers.11.input_layernorm.weight (f16, [2560]): 36%|ββββ | 145/404 [00:05<00:07, 33.03it/s]
Saving model.layers.11.attn.qkv_proj.MatMul.weight (f16, [2560,10240]): 36%|ββββ | 146/404 [00:05<00:07, 33.03it/s]
Saving model.layers.11.attn.q_norm.layernorm.weight (f16, [256]): 36%|ββββ | 147/404 [00:05<00:07, 33.03it/s]
Saving model.layers.11.attn.q_norm.layernorm.weight (f16, [256]): 37%|ββββ | 148/404 [00:05<00:09, 26.06it/s]
Saving model.layers.11.attn.k_norm.layernorm.weight (f16, [256]): 37%|ββββ | 148/404 [00:05<00:09, 26.06it/s]
Saving model.layers.11.attn.o_proj.MatMul.weight (f16, [4096,2560]): 37%|ββββ | 149/404 [00:05<00:09, 26.06it/s]
Saving model.layers.11.post_attention_layernorm.weight (f16, [2560]): 37%|ββββ | 150/404 [00:05<00:09, 26.06it/s]
Saving model.layers.11.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 37%|ββββ | 151/404 [00:05<00:09, 26.06it/s]
Saving model.layers.11.mlp.down_proj.MatMul.weight (f16, [9216,2560]): 38%|ββββ | 152/404 [00:06<00:09, 26.06it/s]
Saving model.layers.11.mlp.down_proj.MatMul.weight (f16, [9216,2560]): 38%|ββββ | 153/404 [00:06<00:11, 21.15it/s]
Saving model.layers.12.input_layernorm.weight (f16, [2560]): 38%|ββββ | 153/404 [00:06<00:11, 21.15it/s]
Saving model.layers.12.linear_attn.qkv_proj.MatMul.weight (f16, [2560,8192]): 38%|ββββ | 154/404 [00:06<00:11, 21.15it/s]
Saving model.layers.12.linear_attn.z_proj.MatMul.weight (f16, [2560,4096]): 38%|ββββ | 155/404 [00:06<00:11, 21.15it/s]
Saving model.layers.12.linear_attn.b_proj.MatMul.weight (f16, [2560,32]): 39%|ββββ | 156/404 [00:06<00:11, 21.15it/s]
Saving model.layers.12.linear_attn.a_proj.MatMul.weight (f16, [2560,32]): 39%|ββββ | 157/404 [00:06<00:11, 21.15it/s]
Saving model.layers.12.linear_attn.conv1d.weight (f16, [8192,1,4]): 39%|ββββ | 158/404 [00:06<00:11, 21.15it/s]
Saving model.layers.12.linear_attn.conv1d.bias (f16, [8192]): 39%|ββββ | 159/404 [00:06<00:11, 21.15it/s]
Saving model.layers.12.linear_attn.A_log (f32, [32]): 40%|ββββ | 160/404 [00:06<00:11, 21.15it/s]
Saving model.layers.12.linear_attn.dt_bias (f32, [32]): 40%|ββββ | 161/404 [00:06<00:11, 21.15it/s]
Saving model.layers.12.linear_attn.norm.weight (f16, [128]): 40%|ββββ | 162/404 [00:06<00:11, 21.15it/s]
Saving model.layers.12.linear_attn.out_proj.MatMul.weight (f16, [4096,2560]): 40%|ββββ | 163/404 [00:06<00:11, 21.15it/s]
Saving model.layers.12.post_attention_layernorm.weight (f16, [2560]): 41%|ββββ | 164/404 [00:06<00:11, 21.15it/s]
Saving model.layers.12.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 41%|ββββ | 165/404 [00:06<00:11, 21.15it/s]
Saving model.layers.12.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 41%|ββββ | 166/404 [00:06<00:10, 23.45it/s]
Saving model.layers.12.mlp.down_proj.MatMul.weight (f16, [9216,2560]): 41%|ββββ | 166/404 [00:06<00:10, 23.45it/s]
Saving model.layers.13.input_layernorm.weight (f16, [2560]): 41%|βββββ | 167/404 [00:06<00:10, 23.45it/s]
Saving model.layers.13.linear_attn.qkv_proj.MatMul.weight (f16, [2560,8192]): 42%|βββββ | 168/404 [00:06<00:10, 23.45it/s]
Saving model.layers.13.linear_attn.z_proj.MatMul.weight (f16, [2560,4096]): 42%|βββββ | 169/404 [00:06<00:10, 23.45it/s]
Saving model.layers.13.linear_attn.b_proj.MatMul.weight (f16, [2560,32]): 42%|βββββ | 170/404 [00:06<00:09, 23.45it/s]
Saving model.layers.13.linear_attn.a_proj.MatMul.weight (f16, [2560,32]): 42%|βββββ | 171/404 [00:06<00:09, 23.45it/s]
Saving model.layers.13.linear_attn.conv1d.weight (f16, [8192,1,4]): 43%|βββββ | 172/404 [00:06<00:09, 23.45it/s]
Saving model.layers.13.linear_attn.conv1d.bias (f16, [8192]): 43%|βββββ | 173/404 [00:06<00:09, 23.45it/s]
Saving model.layers.13.linear_attn.A_log (f32, [32]): 43%|βββββ | 174/404 [00:06<00:09, 23.45it/s]
Saving model.layers.13.linear_attn.dt_bias (f32, [32]): 43%|βββββ | 175/404 [00:06<00:09, 23.45it/s]
Saving model.layers.13.linear_attn.norm.weight (f16, [128]): 44%|βββββ | 176/404 [00:06<00:09, 23.45it/s]
Saving model.layers.13.linear_attn.out_proj.MatMul.weight (f16, [4096,2560]): 44%|βββββ | 177/404 [00:06<00:09, 23.45it/s]
Saving model.layers.13.post_attention_layernorm.weight (f16, [2560]): 44%|βββββ | 178/404 [00:06<00:09, 23.45it/s]
Saving model.layers.13.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 44%|βββββ | 179/404 [00:06<00:09, 23.45it/s]
Saving model.layers.13.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 45%|βββββ | 180/404 [00:06<00:08, 25.09it/s]
Saving model.layers.13.mlp.down_proj.MatMul.weight (f16, [9216,2560]): 45%|βββββ | 180/404 [00:06<00:08, 25.09it/s]
Saving model.layers.14.input_layernorm.weight (f16, [2560]): 45%|βββββ | 181/404 [00:07<00:08, 25.09it/s]
Saving model.layers.14.linear_attn.qkv_proj.MatMul.weight (f16, [2560,8192]): 45%|βββββ | 182/404 [00:07<00:08, 25.09it/s]
Saving model.layers.14.linear_attn.z_proj.MatMul.weight (f16, [2560,4096]): 45%|βββββ | 183/404 [00:07<00:08, 25.09it/s]
Saving model.layers.14.linear_attn.b_proj.MatMul.weight (f16, [2560,32]): 46%|βββββ | 184/404 [00:07<00:08, 25.09it/s]
Saving model.layers.14.linear_attn.a_proj.MatMul.weight (f16, [2560,32]): 46%|βββββ | 185/404 [00:07<00:08, 25.09it/s]
Saving model.layers.14.linear_attn.conv1d.weight (f16, [8192,1,4]): 46%|βββββ | 186/404 [00:07<00:08, 25.09it/s]
Saving model.layers.14.linear_attn.conv1d.bias (f16, [8192]): 46%|βββββ | 187/404 [00:07<00:08, 25.09it/s]
Saving model.layers.14.linear_attn.A_log (f32, [32]): 47%|βββββ | 188/404 [00:07<00:08, 25.09it/s]
Saving model.layers.14.linear_attn.dt_bias (f32, [32]): 47%|βββββ | 189/404 [00:07<00:08, 25.09it/s]
Saving model.layers.14.linear_attn.norm.weight (f16, [128]): 47%|βββββ | 190/404 [00:07<00:08, 25.09it/s]
Saving model.layers.14.linear_attn.out_proj.MatMul.weight (f16, [4096,2560]): 47%|βββββ | 191/404 [00:07<00:08, 25.09it/s]
Saving model.layers.14.post_attention_layernorm.weight (f16, [2560]): 48%|βββββ | 192/404 [00:07<00:08, 25.09it/s]
Saving model.layers.14.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 48%|βββββ | 193/404 [00:07<00:08, 25.09it/s]
Saving model.layers.14.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 48%|βββββ | 194/404 [00:07<00:08, 25.86it/s]
Saving model.layers.14.mlp.down_proj.MatMul.weight (f16, [9216,2560]): 48%|βββββ | 194/404 [00:07<00:08, 25.86it/s]
Saving model.layers.15.input_layernorm.weight (f16, [2560]): 48%|βββββ | 195/404 [00:07<00:08, 25.86it/s]
Saving model.layers.15.attn.qkv_proj.MatMul.weight (f16, [2560,10240]): 49%|βββββ | 196/404 [00:07<00:08, 25.86it/s]
Saving model.layers.15.attn.q_norm.layernorm.weight (f16, [256]): 49%|βββββ | 197/404 [00:07<00:08, 25.86it/s]
Saving model.layers.15.attn.k_norm.layernorm.weight (f16, [256]): 49%|βββββ | 198/404 [00:07<00:07, 25.86it/s]
Saving model.layers.15.attn.o_proj.MatMul.weight (f16, [4096,2560]): 49%|βββββ | 199/404 [00:07<00:07, 25.86it/s]
Saving model.layers.15.post_attention_layernorm.weight (f16, [2560]): 50%|βββββ | 200/404 [00:07<00:07, 25.86it/s]
Saving model.layers.15.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 50%|βββββ | 201/404 [00:07<00:07, 25.86it/s]
Saving model.layers.15.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 50%|βββββ | 202/404 [00:07<00:08, 23.13it/s]
Saving model.layers.15.mlp.down_proj.MatMul.weight (f16, [9216,2560]): 50%|βββββ | 202/404 [00:07<00:08, 23.13it/s]
Saving model.layers.16.input_layernorm.weight (f16, [2560]): 50%|βββββ | 203/404 [00:08<00:08, 23.13it/s]
Saving model.layers.16.linear_attn.qkv_proj.MatMul.weight (f16, [2560,8192]): 50%|βββββ | 204/404 [00:08<00:08, 23.13it/s]
Saving model.layers.16.linear_attn.z_proj.MatMul.weight (f16, [2560,4096]): 51%|βββββ | 205/404 [00:08<00:08, 23.13it/s]
Saving model.layers.16.linear_attn.b_proj.MatMul.weight (f16, [2560,32]): 51%|βββββ | 206/404 [00:08<00:08, 23.13it/s]
Saving model.layers.16.linear_attn.a_proj.MatMul.weight (f16, [2560,32]): 51%|βββββ | 207/404 [00:08<00:08, 23.13it/s]
Saving model.layers.16.linear_attn.conv1d.weight (f16, [8192,1,4]): 51%|ββββββ | 208/404 [00:08<00:08, 23.13it/s]
Saving model.layers.16.linear_attn.conv1d.bias (f16, [8192]): 52%|ββββββ | 209/404 [00:08<00:08, 23.13it/s]
Saving model.layers.16.linear_attn.A_log (f32, [32]): 52%|ββββββ | 210/404 [00:08<00:08, 23.13it/s]
Saving model.layers.16.linear_attn.dt_bias (f32, [32]): 52%|ββββββ | 211/404 [00:08<00:08, 23.13it/s]
Saving model.layers.16.linear_attn.norm.weight (f16, [128]): 52%|ββββββ | 212/404 [00:08<00:08, 23.13it/s]
Saving model.layers.16.linear_attn.out_proj.MatMul.weight (f16, [4096,2560]): 53%|ββββββ | 213/404 [00:08<00:08, 23.13it/s]
Saving model.layers.16.post_attention_layernorm.weight (f16, [2560]): 53%|ββββββ | 214/404 [00:08<00:08, 23.13it/s]
Saving model.layers.16.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 53%|ββββββ | 215/404 [00:08<00:08, 23.13it/s]
Saving model.layers.16.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 53%|ββββββ | 216/404 [00:08<00:07, 24.91it/s]
Saving model.layers.16.mlp.down_proj.MatMul.weight (f16, [9216,2560]): 53%|ββββββ | 216/404 [00:08<00:07, 24.91it/s]
Saving model.layers.17.input_layernorm.weight (f16, [2560]): 54%|ββββββ | 217/404 [00:08<00:07, 24.91it/s]
Saving model.layers.17.linear_attn.qkv_proj.MatMul.weight (f16, [2560,8192]): 54%|ββββββ | 218/404 [00:08<00:07, 24.91it/s]
Saving model.layers.17.linear_attn.z_proj.MatMul.weight (f16, [2560,4096]): 54%|ββββββ | 219/404 [00:08<00:07, 24.91it/s]
Saving model.layers.17.linear_attn.b_proj.MatMul.weight (f16, [2560,32]): 54%|ββββββ | 220/404 [00:08<00:07, 24.91it/s]
Saving model.layers.17.linear_attn.a_proj.MatMul.weight (f16, [2560,32]): 55%|ββββββ | 221/404 [00:08<00:07, 24.91it/s]
Saving model.layers.17.linear_attn.conv1d.weight (f16, [8192,1,4]): 55%|ββββββ | 222/404 [00:08<00:07, 24.91it/s]
Saving model.layers.17.linear_attn.conv1d.bias (f16, [8192]): 55%|ββββββ | 223/404 [00:08<00:07, 24.91it/s]
Saving model.layers.17.linear_attn.A_log (f32, [32]): 55%|ββββββ | 224/404 [00:08<00:07, 24.91it/s]
Saving model.layers.17.linear_attn.dt_bias (f32, [32]): 56%|ββββββ | 225/404 [00:08<00:07, 24.91it/s]
Saving model.layers.17.linear_attn.norm.weight (f16, [128]): 56%|ββββββ | 226/404 [00:08<00:07, 24.91it/s]
Saving model.layers.17.linear_attn.out_proj.MatMul.weight (f16, [4096,2560]): 56%|ββββββ | 227/404 [00:08<00:07, 24.91it/s]
Saving model.layers.17.post_attention_layernorm.weight (f16, [2560]): 56%|ββββββ | 228/404 [00:08<00:07, 24.91it/s]
Saving model.layers.17.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 57%|ββββββ | 229/404 [00:08<00:07, 24.91it/s]
Saving model.layers.17.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 57%|ββββββ | 230/404 [00:08<00:06, 26.16it/s]
Saving model.layers.17.mlp.down_proj.MatMul.weight (f16, [9216,2560]): 57%|ββββββ | 230/404 [00:08<00:06, 26.16it/s]
Saving model.layers.18.input_layernorm.weight (f16, [2560]): 57%|ββββββ | 231/404 [00:08<00:06, 26.16it/s]
Saving model.layers.18.linear_attn.qkv_proj.MatMul.weight (f16, [2560,8192]): 57%|ββββββ | 232/404 [00:08<00:06, 26.16it/s]
Saving model.layers.18.linear_attn.z_proj.MatMul.weight (f16, [2560,4096]): 58%|ββββββ | 233/404 [00:08<00:06, 26.16it/s]
Saving model.layers.18.linear_attn.b_proj.MatMul.weight (f16, [2560,32]): 58%|ββββββ | 234/404 [00:09<00:06, 26.16it/s]
Saving model.layers.18.linear_attn.a_proj.MatMul.weight (f16, [2560,32]): 58%|ββββββ | 235/404 [00:09<00:06, 26.16it/s]
Saving model.layers.18.linear_attn.conv1d.weight (f16, [8192,1,4]): 58%|ββββββ | 236/404 [00:09<00:06, 26.16it/s]
Saving model.layers.18.linear_attn.conv1d.bias (f16, [8192]): 59%|ββββββ | 237/404 [00:09<00:06, 26.16it/s]
Saving model.layers.18.linear_attn.A_log (f32, [32]): 59%|ββββββ | 238/404 [00:09<00:06, 26.16it/s]
Saving model.layers.18.linear_attn.dt_bias (f32, [32]): 59%|ββββββ | 239/404 [00:09<00:06, 26.16it/s]
Saving model.layers.18.linear_attn.norm.weight (f16, [128]): 59%|ββββββ | 240/404 [00:09<00:06, 26.16it/s]
Saving model.layers.18.linear_attn.out_proj.MatMul.weight (f16, [4096,2560]): 60%|ββββββ | 241/404 [00:09<00:06, 26.16it/s]
Saving model.layers.18.post_attention_layernorm.weight (f16, [2560]): 60%|ββββββ | 242/404 [00:09<00:06, 26.16it/s]
Saving model.layers.18.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 60%|ββββββ | 243/404 [00:09<00:06, 26.16it/s]
Saving model.layers.18.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 60%|ββββββ | 244/404 [00:09<00:05, 27.04it/s]
Saving model.layers.18.mlp.down_proj.MatMul.weight (f16, [9216,2560]): 60%|ββββββ | 244/404 [00:09<00:05, 27.04it/s]
Saving model.layers.19.input_layernorm.weight (f16, [2560]): 61%|ββββββ | 245/404 [00:09<00:05, 27.04it/s]
Saving model.layers.19.attn.qkv_proj.MatMul.weight (f16, [2560,10240]): 61%|ββββββ | 246/404 [00:09<00:05, 27.04it/s]
Saving model.layers.19.attn.q_norm.layernorm.weight (f16, [256]): 61%|ββββββ | 247/404 [00:09<00:05, 27.04it/s]
Saving model.layers.19.attn.k_norm.layernorm.weight (f16, [256]): 61%|βββββββ | 248/404 [00:09<00:05, 27.04it/s]
Saving model.layers.19.attn.o_proj.MatMul.weight (f16, [4096,2560]): 62%|βββββββ | 249/404 [00:09<00:05, 27.04it/s]
Saving model.layers.19.post_attention_layernorm.weight (f16, [2560]): 62%|βββββββ | 250/404 [00:09<00:05, 27.04it/s]
Saving model.layers.19.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 62%|βββββββ | 251/404 [00:09<00:05, 27.04it/s]
Saving model.layers.19.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 62%|βββββββ | 252/404 [00:09<00:06, 24.13it/s]
Saving model.layers.19.mlp.down_proj.MatMul.weight (f16, [9216,2560]): 62%|βββββββ | 252/404 [00:09<00:06, 24.13it/s]
Saving model.layers.20.input_layernorm.weight (f16, [2560]): 63%|βββββββ | 253/404 [00:09<00:06, 24.13it/s]
Saving model.layers.20.linear_attn.qkv_proj.MatMul.weight (f16, [2560,8192]): 63%|βββββββ | 254/404 [00:09<00:06, 24.13it/s]
Saving model.layers.20.linear_attn.z_proj.MatMul.weight (f16, [2560,4096]): 63%|βββββββ | 255/404 [00:09<00:06, 24.13it/s]
Saving model.layers.20.linear_attn.b_proj.MatMul.weight (f16, [2560,32]): 63%|βββββββ | 256/404 [00:09<00:06, 24.13it/s]
Saving model.layers.20.linear_attn.a_proj.MatMul.weight (f16, [2560,32]): 64%|βββββββ | 257/404 [00:09<00:06, 24.13it/s]
Saving model.layers.20.linear_attn.conv1d.weight (f16, [8192,1,4]): 64%|βββββββ | 258/404 [00:09<00:06, 24.13it/s]
Saving model.layers.20.linear_attn.conv1d.bias (f16, [8192]): 64%|βββββββ | 259/404 [00:09<00:06, 24.13it/s]
Saving model.layers.20.linear_attn.A_log (f32, [32]): 64%|βββββββ | 260/404 [00:09<00:05, 24.13it/s]
Saving model.layers.20.linear_attn.dt_bias (f32, [32]): 65%|βββββββ | 261/404 [00:09<00:05, 24.13it/s]
Saving model.layers.20.linear_attn.norm.weight (f16, [128]): 65%|βββββββ | 262/404 [00:09<00:05, 24.13it/s]
Saving model.layers.20.linear_attn.out_proj.MatMul.weight (f16, [4096,2560]): 65%|βββββββ | 263/404 [00:09<00:05, 24.13it/s]
Saving model.layers.20.post_attention_layernorm.weight (f16, [2560]): 65%|βββββββ | 264/404 [00:09<00:05, 24.13it/s]
Saving model.layers.20.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 66%|βββββββ | 265/404 [00:09<00:05, 24.13it/s]
Saving model.layers.20.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 66%|βββββββ | 266/404 [00:10<00:05, 25.38it/s]
Saving model.layers.20.mlp.down_proj.MatMul.weight (f16, [9216,2560]): 66%|βββββββ | 266/404 [00:10<00:05, 25.38it/s]
Saving model.layers.21.input_layernorm.weight (f16, [2560]): 66%|βββββββ | 267/404 [00:10<00:05, 25.38it/s]
Saving model.layers.21.linear_attn.qkv_proj.MatMul.weight (f16, [2560,8192]): 66%|βββββββ | 268/404 [00:10<00:05, 25.38it/s]
Saving model.layers.21.linear_attn.z_proj.MatMul.weight (f16, [2560,4096]): 67%|βββββββ | 269/404 [00:10<00:05, 25.38it/s]
Saving model.layers.21.linear_attn.b_proj.MatMul.weight (f16, [2560,32]): 67%|βββββββ | 270/404 [00:10<00:05, 25.38it/s]
Saving model.layers.21.linear_attn.a_proj.MatMul.weight (f16, [2560,32]): 67%|βββββββ | 271/404 [00:10<00:05, 25.38it/s]
Saving model.layers.21.linear_attn.conv1d.weight (f16, [8192,1,4]): 67%|βββββββ | 272/404 [00:10<00:05, 25.38it/s]
Saving model.layers.21.linear_attn.conv1d.bias (f16, [8192]): 68%|βββββββ | 273/404 [00:10<00:05, 25.38it/s]
Saving model.layers.21.linear_attn.A_log (f32, [32]): 68%|βββββββ | 274/404 [00:10<00:05, 25.38it/s]
Saving model.layers.21.linear_attn.dt_bias (f32, [32]): 68%|βββββββ | 275/404 [00:10<00:05, 25.38it/s]
Saving model.layers.21.linear_attn.norm.weight (f16, [128]): 68%|βββββββ | 276/404 [00:10<00:05, 25.38it/s]
Saving model.layers.21.linear_attn.out_proj.MatMul.weight (f16, [4096,2560]): 69%|βββββββ | 277/404 [00:10<00:05, 25.38it/s]
Saving model.layers.21.post_attention_layernorm.weight (f16, [2560]): 69%|βββββββ | 278/404 [00:10<00:04, 25.38it/s]
Saving model.layers.21.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 69%|βββββββ | 279/404 [00:10<00:04, 25.38it/s]
Saving model.layers.21.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 69%|βββββββ | 280/404 [00:10<00:04, 26.50it/s]
Saving model.layers.21.mlp.down_proj.MatMul.weight (f16, [9216,2560]): 69%|βββββββ | 280/404 [00:10<00:04, 26.50it/s]
Saving model.layers.22.input_layernorm.weight (f16, [2560]): 70%|βββββββ | 281/404 [00:10<00:04, 26.50it/s]
Saving model.layers.22.linear_attn.qkv_proj.MatMul.weight (f16, [2560,8192]): 70%|βββββββ | 282/404 [00:10<00:04, 26.50it/s]
Saving model.layers.22.linear_attn.z_proj.MatMul.weight (f16, [2560,4096]): 70%|βββββββ | 283/404 [00:10<00:04, 26.50it/s]
Saving model.layers.22.linear_attn.b_proj.MatMul.weight (f16, [2560,32]): 70%|βββββββ | 284/404 [00:10<00:04, 26.50it/s]
Saving model.layers.22.linear_attn.a_proj.MatMul.weight (f16, [2560,32]): 71%|βββββββ | 285/404 [00:10<00:04, 26.50it/s]
Saving model.layers.22.linear_attn.conv1d.weight (f16, [8192,1,4]): 71%|βββββββ | 286/404 [00:10<00:04, 26.50it/s]
Saving model.layers.22.linear_attn.conv1d.bias (f16, [8192]): 71%|βββββββ | 287/404 [00:10<00:04, 26.50it/s]
Saving model.layers.22.linear_attn.A_log (f32, [32]): 71%|ββββββββ | 288/404 [00:10<00:04, 26.50it/s]
Saving model.layers.22.linear_attn.dt_bias (f32, [32]): 72%|ββββββββ | 289/404 [00:10<00:04, 26.50it/s]
Saving model.layers.22.linear_attn.norm.weight (f16, [128]): 72%|ββββββββ | 290/404 [00:10<00:04, 26.50it/s]
Saving model.layers.22.linear_attn.out_proj.MatMul.weight (f16, [4096,2560]): 72%|ββββββββ | 291/404 [00:10<00:04, 26.50it/s]
Saving model.layers.22.post_attention_layernorm.weight (f16, [2560]): 72%|ββββββββ | 292/404 [00:10<00:04, 26.50it/s]
Saving model.layers.22.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 73%|ββββββββ | 293/404 [00:10<00:04, 26.50it/s]
Saving model.layers.22.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 73%|ββββββββ | 294/404 [00:11<00:03, 27.68it/s]
Saving model.layers.22.mlp.down_proj.MatMul.weight (f16, [9216,2560]): 73%|ββββββββ | 294/404 [00:11<00:03, 27.68it/s]
Saving model.layers.23.input_layernorm.weight (f16, [2560]): 73%|ββββββββ | 295/404 [00:11<00:03, 27.68it/s]
Saving model.layers.23.attn.qkv_proj.MatMul.weight (f16, [2560,10240]): 73%|ββββββββ | 296/404 [00:11<00:03, 27.68it/s]
Saving model.layers.23.attn.q_norm.layernorm.weight (f16, [256]): 74%|ββββββββ | 297/404 [00:11<00:03, 27.68it/s]
Saving model.layers.23.attn.k_norm.layernorm.weight (f16, [256]): 74%|ββββββββ | 298/404 [00:11<00:03, 27.68it/s]
Saving model.layers.23.attn.o_proj.MatMul.weight (f16, [4096,2560]): 74%|ββββββββ | 299/404 [00:11<00:03, 27.68it/s]
Saving model.layers.23.post_attention_layernorm.weight (f16, [2560]): 74%|ββββββββ | 300/404 [00:11<00:03, 27.68it/s]
Saving model.layers.23.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 75%|ββββββββ | 301/404 [00:11<00:03, 27.68it/s]
Saving model.layers.23.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 75%|ββββββββ | 302/404 [00:11<00:04, 24.59it/s]
Saving model.layers.23.mlp.down_proj.MatMul.weight (f16, [9216,2560]): 75%|ββββββββ | 302/404 [00:11<00:04, 24.59it/s]
Saving model.layers.24.input_layernorm.weight (f16, [2560]): 75%|ββββββββ | 303/404 [00:11<00:04, 24.59it/s]
Saving model.layers.24.linear_attn.qkv_proj.MatMul.weight (f16, [2560,8192]): 75%|ββββββββ | 304/404 [00:11<00:04, 24.59it/s]
Saving model.layers.24.linear_attn.z_proj.MatMul.weight (f16, [2560,4096]): 75%|ββββββββ | 305/404 [00:11<00:04, 24.59it/s]
Saving model.layers.24.linear_attn.b_proj.MatMul.weight (f16, [2560,32]): 76%|ββββββββ | 306/404 [00:11<00:03, 24.59it/s]
Saving model.layers.24.linear_attn.a_proj.MatMul.weight (f16, [2560,32]): 76%|ββββββββ | 307/404 [00:11<00:03, 24.59it/s]
Saving model.layers.24.linear_attn.conv1d.weight (f16, [8192,1,4]): 76%|ββββββββ | 308/404 [00:11<00:03, 24.59it/s]
Saving model.layers.24.linear_attn.conv1d.bias (f16, [8192]): 76%|ββββββββ | 309/404 [00:11<00:03, 24.59it/s]
Saving model.layers.24.linear_attn.A_log (f32, [32]): 77%|ββββββββ | 310/404 [00:11<00:03, 24.59it/s]
Saving model.layers.24.linear_attn.dt_bias (f32, [32]): 77%|ββββββββ | 311/404 [00:11<00:03, 24.59it/s]
Saving model.layers.24.linear_attn.norm.weight (f16, [128]): 77%|ββββββββ | 312/404 [00:11<00:03, 24.59it/s]
Saving model.layers.24.linear_attn.out_proj.MatMul.weight (f16, [4096,2560]): 77%|ββββββββ | 313/404 [00:11<00:03, 24.59it/s]
Saving model.layers.24.post_attention_layernorm.weight (f16, [2560]): 78%|ββββββββ | 314/404 [00:11<00:03, 24.59it/s]
Saving model.layers.24.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 78%|ββββββββ | 315/404 [00:11<00:03, 24.59it/s]
Saving model.layers.24.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 78%|ββββββββ | 316/404 [00:12<00:03, 25.91it/s]
Saving model.layers.24.mlp.down_proj.MatMul.weight (f16, [9216,2560]): 78%|ββββββββ | 316/404 [00:12<00:03, 25.91it/s]
Saving model.layers.25.input_layernorm.weight (f16, [2560]): 78%|ββββββββ | 317/404 [00:12<00:03, 25.91it/s]
Saving model.layers.25.linear_attn.qkv_proj.MatMul.weight (f16, [2560,8192]): 79%|ββββββββ | 318/404 [00:12<00:03, 25.91it/s]
Saving model.layers.25.linear_attn.z_proj.MatMul.weight (f16, [2560,4096]): 79%|ββββββββ | 319/404 [00:12<00:03, 25.91it/s]
Saving model.layers.25.linear_attn.b_proj.MatMul.weight (f16, [2560,32]): 79%|ββββββββ | 320/404 [00:12<00:03, 25.91it/s]
Saving model.layers.25.linear_attn.a_proj.MatMul.weight (f16, [2560,32]): 79%|ββββββββ | 321/404 [00:12<00:03, 25.91it/s]
Saving model.layers.25.linear_attn.conv1d.weight (f16, [8192,1,4]): 80%|ββββββββ | 322/404 [00:12<00:03, 25.91it/s]
Saving model.layers.25.linear_attn.conv1d.bias (f16, [8192]): 80%|ββββββββ | 323/404 [00:12<00:03, 25.91it/s]
Saving model.layers.25.linear_attn.A_log (f32, [32]): 80%|ββββββββ | 324/404 [00:12<00:03, 25.91it/s]
Saving model.layers.25.linear_attn.dt_bias (f32, [32]): 80%|ββββββββ | 325/404 [00:12<00:03, 25.91it/s]
Saving model.layers.25.linear_attn.norm.weight (f16, [128]): 81%|ββββββββ | 326/404 [00:12<00:03, 25.91it/s]
Saving model.layers.25.linear_attn.out_proj.MatMul.weight (f16, [4096,2560]): 81%|ββββββββ | 327/404 [00:12<00:02, 25.91it/s]
Saving model.layers.25.post_attention_layernorm.weight (f16, [2560]): 81%|ββββββββ | 328/404 [00:12<00:02, 25.91it/s]
Saving model.layers.25.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 81%|βββββββββ | 329/404 [00:12<00:02, 25.91it/s]
Saving model.layers.25.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 82%|βββββββββ | 330/404 [00:12<00:02, 26.86it/s]
Saving model.layers.25.mlp.down_proj.MatMul.weight (f16, [9216,2560]): 82%|βββββββββ | 330/404 [00:12<00:02, 26.86it/s]
Saving model.layers.26.input_layernorm.weight (f16, [2560]): 82%|βββββββββ | 331/404 [00:12<00:02, 26.86it/s]
Saving model.layers.26.linear_attn.qkv_proj.MatMul.weight (f16, [2560,8192]): 82%|βββββββββ | 332/404 [00:12<00:02, 26.86it/s]
Saving model.layers.26.linear_attn.z_proj.MatMul.weight (f16, [2560,4096]): 82%|βββββββββ | 333/404 [00:12<00:02, 26.86it/s]
Saving model.layers.26.linear_attn.b_proj.MatMul.weight (f16, [2560,32]): 83%|βββββββββ | 334/404 [00:12<00:02, 26.86it/s]
Saving model.layers.26.linear_attn.a_proj.MatMul.weight (f16, [2560,32]): 83%|βββββββββ | 335/404 [00:12<00:02, 26.86it/s]
Saving model.layers.26.linear_attn.conv1d.weight (f16, [8192,1,4]): 83%|βββββββββ | 336/404 [00:12<00:02, 26.86it/s]
Saving model.layers.26.linear_attn.conv1d.bias (f16, [8192]): 83%|βββββββββ | 337/404 [00:12<00:02, 26.86it/s]
Saving model.layers.26.linear_attn.A_log (f32, [32]): 84%|βββββββββ | 338/404 [00:12<00:02, 26.86it/s]
Saving model.layers.26.linear_attn.dt_bias (f32, [32]): 84%|βββββββββ | 339/404 [00:12<00:02, 26.86it/s]
Saving model.layers.26.linear_attn.norm.weight (f16, [128]): 84%|βββββββββ | 340/404 [00:12<00:02, 26.86it/s]
Saving model.layers.26.linear_attn.out_proj.MatMul.weight (f16, [4096,2560]): 84%|βββββββββ | 341/404 [00:12<00:02, 26.86it/s]
Saving model.layers.26.post_attention_layernorm.weight (f16, [2560]): 85%|βββββββββ | 342/404 [00:12<00:02, 26.86it/s]
Saving model.layers.26.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 85%|βββββββββ | 343/404 [00:12<00:02, 26.86it/s]
Saving model.layers.26.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 85%|βββββββββ | 344/404 [00:13<00:02, 27.08it/s]
Saving model.layers.26.mlp.down_proj.MatMul.weight (f16, [9216,2560]): 85%|βββββββββ | 344/404 [00:13<00:02, 27.08it/s]
Saving model.layers.27.input_layernorm.weight (f16, [2560]): 85%|βββββββββ | 345/404 [00:13<00:02, 27.08it/s]
Saving model.layers.27.attn.qkv_proj.MatMul.weight (f16, [2560,10240]): 86%|βββββββββ | 346/404 [00:13<00:02, 27.08it/s]
Saving model.layers.27.attn.q_norm.layernorm.weight (f16, [256]): 86%|βββββββββ | 347/404 [00:13<00:02, 27.08it/s]
Saving model.layers.27.attn.k_norm.layernorm.weight (f16, [256]): 86%|βββββββββ | 348/404 [00:13<00:02, 27.08it/s]
Saving model.layers.27.attn.o_proj.MatMul.weight (f16, [4096,2560]): 86%|βββββββββ | 349/404 [00:13<00:02, 27.08it/s]
Saving model.layers.27.post_attention_layernorm.weight (f16, [2560]): 87%|βββββββββ | 350/404 [00:13<00:01, 27.08it/s]
Saving model.layers.27.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 87%|βββββββββ | 351/404 [00:13<00:01, 27.08it/s]
Saving model.layers.27.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 87%|βββββββββ | 352/404 [00:13<00:02, 23.82it/s]
Saving model.layers.27.mlp.down_proj.MatMul.weight (f16, [9216,2560]): 87%|βββββββββ | 352/404 [00:13<00:02, 23.82it/s]
Saving model.layers.28.input_layernorm.weight (f16, [2560]): 87%|βββββββββ | 353/404 [00:13<00:02, 23.82it/s]
Saving model.layers.28.linear_attn.qkv_proj.MatMul.weight (f16, [2560,8192]): 88%|βββββββββ | 354/404 [00:13<00:02, 23.82it/s]
Saving model.layers.28.linear_attn.z_proj.MatMul.weight (f16, [2560,4096]): 88%|βββββββββ | 355/404 [00:13<00:02, 23.82it/s]
Saving model.layers.28.linear_attn.b_proj.MatMul.weight (f16, [2560,32]): 88%|βββββββββ | 356/404 [00:13<00:02, 23.82it/s]
Saving model.layers.28.linear_attn.a_proj.MatMul.weight (f16, [2560,32]): 88%|βββββββββ | 357/404 [00:13<00:01, 23.82it/s]
Saving model.layers.28.linear_attn.conv1d.weight (f16, [8192,1,4]): 89%|βββββββββ | 358/404 [00:13<00:01, 23.82it/s]
Saving model.layers.28.linear_attn.conv1d.bias (f16, [8192]): 89%|βββββββββ | 359/404 [00:13<00:01, 23.82it/s]
Saving model.layers.28.linear_attn.A_log (f32, [32]): 89%|βββββββββ | 360/404 [00:13<00:01, 23.82it/s]
Saving model.layers.28.linear_attn.dt_bias (f32, [32]): 89%|βββββββββ | 361/404 [00:13<00:01, 23.82it/s]
Saving model.layers.28.linear_attn.norm.weight (f16, [128]): 90%|βββββββββ | 362/404 [00:13<00:01, 23.82it/s]
Saving model.layers.28.linear_attn.out_proj.MatMul.weight (f16, [4096,2560]): 90%|βββββββββ | 363/404 [00:13<00:01, 23.82it/s]
Saving model.layers.28.post_attention_layernorm.weight (f16, [2560]): 90%|βββββββββ | 364/404 [00:13<00:01, 23.82it/s]
Saving model.layers.28.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 90%|βββββββββ | 365/404 [00:13<00:01, 23.82it/s]
Saving model.layers.28.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 91%|βββββββββ | 366/404 [00:14<00:01, 25.41it/s]
Saving model.layers.28.mlp.down_proj.MatMul.weight (f16, [9216,2560]): 91%|βββββββββ | 366/404 [00:14<00:01, 25.41it/s]
Saving model.layers.29.input_layernorm.weight (f16, [2560]): 91%|βββββββββ | 367/404 [00:14<00:01, 25.41it/s]
Saving model.layers.29.linear_attn.qkv_proj.MatMul.weight (f16, [2560,8192]): 91%|βββββββββ | 368/404 [00:14<00:01, 25.41it/s]
Saving model.layers.29.linear_attn.z_proj.MatMul.weight (f16, [2560,4096]): 91%|ββββββββββ| 369/404 [00:14<00:01, 25.41it/s]
Saving model.layers.29.linear_attn.b_proj.MatMul.weight (f16, [2560,32]): 92%|ββββββββββ| 370/404 [00:14<00:01, 25.41it/s]
Saving model.layers.29.linear_attn.a_proj.MatMul.weight (f16, [2560,32]): 92%|ββββββββββ| 371/404 [00:14<00:01, 25.41it/s]
Saving model.layers.29.linear_attn.conv1d.weight (f16, [8192,1,4]): 92%|ββββββββββ| 372/404 [00:14<00:01, 25.41it/s]
Saving model.layers.29.linear_attn.conv1d.bias (f16, [8192]): 92%|ββββββββββ| 373/404 [00:14<00:01, 25.41it/s]
Saving model.layers.29.linear_attn.A_log (f32, [32]): 93%|ββββββββββ| 374/404 [00:14<00:01, 25.41it/s]
Saving model.layers.29.linear_attn.dt_bias (f32, [32]): 93%|ββββββββββ| 375/404 [00:14<00:01, 25.41it/s]
Saving model.layers.29.linear_attn.norm.weight (f16, [128]): 93%|ββββββββββ| 376/404 [00:14<00:01, 25.41it/s]
Saving model.layers.29.linear_attn.out_proj.MatMul.weight (f16, [4096,2560]): 93%|ββββββββββ| 377/404 [00:14<00:01, 25.41it/s]
Saving model.layers.29.post_attention_layernorm.weight (f16, [2560]): 94%|ββββββββββ| 378/404 [00:14<00:01, 25.41it/s]
Saving model.layers.29.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 94%|ββββββββββ| 379/404 [00:14<00:00, 25.41it/s]
Saving model.layers.29.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 94%|ββββββββββ| 380/404 [00:14<00:00, 26.08it/s]
Saving model.layers.29.mlp.down_proj.MatMul.weight (f16, [9216,2560]): 94%|ββββββββββ| 380/404 [00:14<00:00, 26.08it/s]
Saving model.layers.30.input_layernorm.weight (f16, [2560]): 94%|ββββββββββ| 381/404 [00:14<00:00, 26.08it/s]
Saving model.layers.30.linear_attn.qkv_proj.MatMul.weight (f16, [2560,8192]): 95%|ββββββββββ| 382/404 [00:14<00:00, 26.08it/s]
Saving model.layers.30.linear_attn.z_proj.MatMul.weight (f16, [2560,4096]): 95%|ββββββββββ| 383/404 [00:14<00:00, 26.08it/s]
Saving model.layers.30.linear_attn.b_proj.MatMul.weight (f16, [2560,32]): 95%|ββββββββββ| 384/404 [00:14<00:00, 26.08it/s]
Saving model.layers.30.linear_attn.a_proj.MatMul.weight (f16, [2560,32]): 95%|ββββββββββ| 385/404 [00:14<00:00, 26.08it/s]
Saving model.layers.30.linear_attn.conv1d.weight (f16, [8192,1,4]): 96%|ββββββββββ| 386/404 [00:14<00:00, 26.08it/s]
Saving model.layers.30.linear_attn.conv1d.bias (f16, [8192]): 96%|ββββββββββ| 387/404 [00:14<00:00, 26.08it/s]
Saving model.layers.30.linear_attn.A_log (f32, [32]): 96%|ββββββββββ| 388/404 [00:14<00:00, 26.08it/s]
Saving model.layers.30.linear_attn.dt_bias (f32, [32]): 96%|ββββββββββ| 389/404 [00:14<00:00, 26.08it/s]
Saving model.layers.30.linear_attn.norm.weight (f16, [128]): 97%|ββββββββββ| 390/404 [00:14<00:00, 26.08it/s]
Saving model.layers.30.linear_attn.out_proj.MatMul.weight (f16, [4096,2560]): 97%|ββββββββββ| 391/404 [00:14<00:00, 26.08it/s]
Saving model.layers.30.post_attention_layernorm.weight (f16, [2560]): 97%|ββββββββββ| 392/404 [00:14<00:00, 26.08it/s]
Saving model.layers.30.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 97%|ββββββββββ| 393/404 [00:14<00:00, 26.08it/s]
Saving model.layers.30.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 98%|ββββββββββ| 394/404 [00:15<00:00, 25.98it/s]
Saving model.layers.30.mlp.down_proj.MatMul.weight (f16, [9216,2560]): 98%|ββββββββββ| 394/404 [00:15<00:00, 25.98it/s]
Saving model.layers.31.input_layernorm.weight (f16, [2560]): 98%|ββββββββββ| 395/404 [00:15<00:00, 25.98it/s]
Saving model.layers.31.attn.qkv_proj.MatMul.weight (f16, [2560,10240]): 98%|ββββββββββ| 396/404 [00:15<00:00, 25.98it/s]
Saving model.layers.31.attn.q_norm.layernorm.weight (f16, [256]): 98%|ββββββββββ| 397/404 [00:15<00:00, 25.98it/s]
Saving model.layers.31.attn.k_norm.layernorm.weight (f16, [256]): 99%|ββββββββββ| 398/404 [00:15<00:00, 25.98it/s]
Saving model.layers.31.attn.o_proj.MatMul.weight (f16, [4096,2560]): 99%|ββββββββββ| 399/404 [00:15<00:00, 25.98it/s]
Saving model.layers.31.post_attention_layernorm.weight (f16, [2560]): 99%|ββββββββββ| 400/404 [00:15<00:00, 25.98it/s]
Saving model.layers.31.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 99%|ββββββββββ| 401/404 [00:15<00:00, 25.98it/s]
Saving model.layers.31.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 100%|ββββββββββ| 402/404 [00:15<00:00, 21.55it/s]
Saving model.layers.31.mlp.down_proj.MatMul.weight (f16, [9216,2560]): 100%|ββββββββββ| 402/404 [00:15<00:00, 21.55it/s]
Saving model.layers.32.final_norm_layernorm.weight (f16, [2560]): 100%|ββββββββββ| 403/404 [00:15<00:00, 21.55it/s]
Saving lm_head.MatMul.weight (f16, [2560,248320]): 100%|ββββββββββ| 404/404 [00:15<00:00, 21.55it/s]
Saving lm_head.MatMul.weight (f16, [2560,248320]): 100%|ββββββββββ| 404/404 [00:17<00:00, 23.52it/s]
Saving ONNX model in /home/anon/knowline/onnx/model
0it [00:00, ?it/s]
Saving model.mtp.pre_fc_norm_embedding.weight (f16, [2560]): 7%|β | 1/15 [00:00<00:00, 678.47it/s]
Saving model.mtp.pre_fc_norm_hidden.weight (f16, [2560]): 13%|ββ | 2/15 [00:00<00:00, 1191.90it/s]
Saving model.mtp.fc.MatMul.weight (f16, [5120,2560]): 20%|ββ | 3/15 [00:00<00:00, 1704.08it/s]
Saving model.layers.0.input_layernorm.weight (f16, [2560]): 27%|βββ | 4/15 [00:00<00:00, 183.98it/s]
Saving model.layers.0.attn.qkv_proj.MatMul.weight (f16, [2560,10240]): 33%|ββββ | 5/15 [00:00<00:00, 227.77it/s]
Saving model.layers.0.attn.q_norm.layernorm.weight (f16, [256]): 40%|ββββ | 6/15 [00:00<00:00, 108.10it/s]
Saving model.layers.0.attn.k_norm.layernorm.weight (f16, [256]): 47%|βββββ | 7/15 [00:00<00:00, 120.96it/s]
Saving cos_cache (f32, [262144,32]): 53%|ββββββ | 8/15 [00:00<00:00, 137.95it/s]
Saving sin_cache (f32, [262144,32]): 60%|ββββββ | 9/15 [00:00<00:00, 126.02it/s]
Saving model.layers.0.attn.o_proj.MatMul.weight (f16, [4096,2560]): 67%|βββββββ | 10/15 [00:00<00:00, 125.93it/s]
Saving model.layers.0.attn.o_proj.MatMul.weight (f16, [4096,2560]): 73%|ββββββββ | 11/15 [00:00<00:00, 104.02it/s]
Saving model.layers.0.post_attention_layernorm.weight (f16, [2560]): 73%|ββββββββ | 11/15 [00:00<00:00, 104.02it/s]
Saving model.layers.0.mlp.gate_up_proj.MatMul.input_1 (f16, [2560,18432]): 80%|ββββββββ | 12/15 [00:00<00:00, 104.02it/s]
Saving model.layers.0.mlp.down_proj.MatMul.weight (f16, [9216,2560]): 87%|βββββββββ | 13/15 [00:00<00:00, 104.02it/s]
Saving model.layers.1.final_norm_layernorm.weight (f16, [2560]): 93%|ββββββββββ| 14/15 [00:00<00:00, 104.02it/s]
Saving lm_head.MatMul.weight (f16, [2560,248320]): 100%|ββββββββββ| 15/15 [00:00<00:00, 104.02it/s]
Saving lm_head.MatMul.weight (f16, [2560,248320]): 100%|ββββββββββ| 15/15 [00:01<00:00, 9.50it/s]
Configured 1271 MB of runtime initializer sharing; /home/anon/knowline/onnx/model/mtp.onnx.data remains self-contained.
Warning: could not read generation_config.json (/home/anon/LLM/KnowLine-4B-Gen5.5 does not appear to have a file named generation_config.json. Checkout 'https://huggingface.co//home/anon/LLM/KnowLine-4B-Gen5.5/tree/main' for available files.). Falling back to config.json.
Adding the tokenizer's end-of-turn token <|im_end|> (id 248046) to eos_token_id from the model configuration: 248044.
Saving GenAI config in /home/anon/knowline/onnx/model
Added 'mtp' section to genai_config.json
Saving processing files in /home/anon/knowline/onnx/model for GenAI
EXIT:0
|