|
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] |