APEX FusedRMSNorm not available, using native implementation
Loading base + LoRA...
Tied input and output embeddings using standard assignment.

Loading checkpoint shards:   0%|          | 0/3 [00:00<?, ?it/s]
Loading checkpoint shards:  67%|██████▋   | 2/3 [00:00<00:00, 10.87it/s]
Loading checkpoint shards: 100%|██████████| 3/3 [00:00<00:00,  8.09it/s]
Tied input and output embeddings using standard assignment.
/home/ubuntu/.local/lib/python3.10/site-packages/peft/peft_model.py:598: UserWarning: Found missing adapter keys while loading the checkpoint: ['base_model.model.model.language_model.layers.0.self_attn.q_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.0.self_attn.q_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.0.self_attn.k_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.0.self_attn.k_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.0.self_attn.v_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.0.self_attn.v_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.0.self_attn.o_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.0.self_attn.o_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.0.mlp.gate_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.0.mlp.gate_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.0.mlp.up_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.0.mlp.up_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.0.mlp.down_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.0.mlp.down_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.1.self_attn.q_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.1.self_attn.q_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.1.self_attn.k_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.1.self_attn.k_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.1.self_attn.v_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.1.self_attn.v_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.1.self_attn.o_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.1.self_attn.o_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.1.mlp.gate_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.1.mlp.gate_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.1.mlp.up_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.1.mlp.up_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.1.mlp.down_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.1.mlp.down_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.2.self_attn.q_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.2.self_attn.q_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.2.self_attn.k_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.2.self_attn.k_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.2.self_attn.v_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.2.self_attn.v_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.2.self_attn.o_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.2.self_attn.o_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.2.mlp.gate_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.2.mlp.gate_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.2.mlp.up_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.2.mlp.up_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.2.mlp.down_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.2.mlp.down_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.3.self_attn.q_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.3.self_attn.q_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.3.self_attn.k_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.3.self_attn.k_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.3.self_attn.v_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.3.self_attn.v_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.3.self_attn.o_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.3.self_attn.o_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.3.mlp.gate_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.3.mlp.gate_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.3.mlp.up_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.3.mlp.up_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.3.mlp.down_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.3.mlp.down_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.4.self_attn.q_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.4.self_attn.q_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.4.self_attn.k_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.4.self_attn.k_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.4.self_attn.v_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.4.self_attn.v_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.4.self_attn.o_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.4.self_attn.o_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.4.mlp.gate_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.4.mlp.gate_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.4.mlp.up_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.4.mlp.up_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.4.mlp.down_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.4.mlp.down_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.5.self_attn.q_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.5.self_attn.q_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.5.self_attn.k_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.5.self_attn.k_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.5.self_attn.v_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.5.self_attn.v_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.5.self_attn.o_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.5.self_attn.o_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.5.mlp.gate_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.5.mlp.gate_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.5.mlp.up_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.5.mlp.up_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.5.mlp.down_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.5.mlp.down_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.6.self_attn.q_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.6.self_attn.q_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.6.self_attn.k_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.6.self_attn.k_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.6.self_attn.v_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.6.self_attn.v_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.6.self_attn.o_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.6.self_attn.o_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.6.mlp.gate_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.6.mlp.gate_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.6.mlp.up_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.6.mlp.up_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.6.mlp.down_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.6.mlp.down_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.7.self_attn.q_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.7.self_attn.q_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.7.self_attn.k_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.7.self_attn.k_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.7.self_attn.v_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.7.self_attn.v_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.7.self_attn.o_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.7.self_attn.o_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.7.mlp.gate_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.7.mlp.gate_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.7.mlp.up_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.7.mlp.up_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.7.mlp.down_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.7.mlp.down_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.8.self_attn.q_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.8.self_attn.q_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.8.self_attn.k_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.8.self_attn.k_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.8.self_attn.v_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.8.self_attn.v_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.8.self_attn.o_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.8.self_attn.o_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.8.mlp.gate_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.8.mlp.gate_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.8.mlp.up_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.8.mlp.up_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.8.mlp.down_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.8.mlp.down_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.9.self_attn.q_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.9.self_attn.q_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.9.self_attn.k_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.9.self_attn.k_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.9.self_attn.v_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.9.self_attn.v_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.9.self_attn.o_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.9.self_attn.o_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.9.mlp.gate_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.9.mlp.gate_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.9.mlp.up_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.9.mlp.up_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.9.mlp.down_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.9.mlp.down_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.10.self_attn.q_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.10.self_attn.q_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.10.self_attn.k_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.10.self_attn.k_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.10.self_attn.v_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.10.self_attn.v_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.10.self_attn.o_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.10.self_attn.o_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.10.mlp.gate_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.10.mlp.gate_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.10.mlp.up_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.10.mlp.up_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.10.mlp.down_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.10.mlp.down_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.11.self_attn.q_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.11.self_attn.q_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.11.self_attn.k_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.11.self_attn.k_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.11.self_attn.v_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.11.self_attn.v_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.11.self_attn.o_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.11.self_attn.o_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.11.mlp.gate_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.11.mlp.gate_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.11.mlp.up_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.11.mlp.up_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.11.mlp.down_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.11.mlp.down_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.12.self_attn.q_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.12.self_attn.q_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.12.self_attn.k_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.12.self_attn.k_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.12.self_attn.v_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.12.self_attn.v_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.12.self_attn.o_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.12.self_attn.o_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.12.mlp.gate_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.12.mlp.gate_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.12.mlp.up_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.12.mlp.up_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.12.mlp.down_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.12.mlp.down_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.13.self_attn.q_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.13.self_attn.q_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.13.self_attn.k_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.13.self_attn.k_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.13.self_attn.v_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.13.self_attn.v_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.13.self_attn.o_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.13.self_attn.o_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.13.mlp.gate_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.13.mlp.gate_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.13.mlp.up_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.13.mlp.up_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.13.mlp.down_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.13.mlp.down_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.14.self_attn.q_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.14.self_attn.q_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.14.self_attn.k_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.14.self_attn.k_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.14.self_attn.v_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.14.self_attn.v_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.14.self_attn.o_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.14.self_attn.o_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.14.mlp.gate_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.14.mlp.gate_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.14.mlp.up_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.14.mlp.up_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.14.mlp.down_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.14.mlp.down_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.15.self_attn.q_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.15.self_attn.q_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.15.self_attn.k_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.15.self_attn.k_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.15.self_attn.v_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.15.self_attn.v_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.15.self_attn.o_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.15.self_attn.o_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.15.mlp.gate_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.15.mlp.gate_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.15.mlp.up_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.15.mlp.up_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.15.mlp.down_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.15.mlp.down_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.16.self_attn.q_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.16.self_attn.q_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.16.self_attn.k_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.16.self_attn.k_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.16.self_attn.v_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.16.self_attn.v_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.16.self_attn.o_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.16.self_attn.o_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.16.mlp.gate_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.16.mlp.gate_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.16.mlp.up_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.16.mlp.up_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.16.mlp.down_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.16.mlp.down_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.17.self_attn.q_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.17.self_attn.q_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.17.self_attn.k_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.17.self_attn.k_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.17.self_attn.v_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.17.self_attn.v_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.17.self_attn.o_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.17.self_attn.o_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.17.mlp.gate_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.17.mlp.gate_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.17.mlp.up_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.17.mlp.up_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.17.mlp.down_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.17.mlp.down_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.18.self_attn.q_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.18.self_attn.q_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.18.self_attn.k_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.18.self_attn.k_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.18.self_attn.v_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.18.self_attn.v_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.18.self_attn.o_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.18.self_attn.o_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.18.mlp.gate_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.18.mlp.gate_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.18.mlp.up_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.18.mlp.up_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.18.mlp.down_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.18.mlp.down_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.19.self_attn.q_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.19.self_attn.q_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.19.self_attn.k_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.19.self_attn.k_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.19.self_attn.v_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.19.self_attn.v_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.19.self_attn.o_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.19.self_attn.o_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.19.mlp.gate_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.19.mlp.gate_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.19.mlp.up_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.19.mlp.up_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.19.mlp.down_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.19.mlp.down_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.20.self_attn.q_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.20.self_attn.q_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.20.self_attn.k_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.20.self_attn.k_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.20.self_attn.v_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.20.self_attn.v_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.20.self_attn.o_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.20.self_attn.o_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.20.mlp.gate_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.20.mlp.gate_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.20.mlp.up_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.20.mlp.up_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.20.mlp.down_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.20.mlp.down_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.21.self_attn.q_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.21.self_attn.q_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.21.self_attn.k_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.21.self_attn.k_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.21.self_attn.v_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.21.self_attn.v_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.21.self_attn.o_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.21.self_attn.o_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.21.mlp.gate_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.21.mlp.gate_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.21.mlp.up_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.21.mlp.up_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.21.mlp.down_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.21.mlp.down_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.22.self_attn.q_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.22.self_attn.q_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.22.self_attn.k_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.22.self_attn.k_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.22.self_attn.v_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.22.self_attn.v_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.22.self_attn.o_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.22.self_attn.o_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.22.mlp.gate_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.22.mlp.gate_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.22.mlp.up_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.22.mlp.up_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.22.mlp.down_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.22.mlp.down_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.23.self_attn.q_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.23.self_attn.q_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.23.self_attn.k_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.23.self_attn.k_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.23.self_attn.v_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.23.self_attn.v_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.23.self_attn.o_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.23.self_attn.o_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.23.mlp.gate_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.23.mlp.gate_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.23.mlp.up_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.23.mlp.up_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.23.mlp.down_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.23.mlp.down_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.24.self_attn.q_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.24.self_attn.q_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.24.self_attn.k_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.24.self_attn.k_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.24.self_attn.v_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.24.self_attn.v_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.24.self_attn.o_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.24.self_attn.o_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.24.mlp.gate_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.24.mlp.gate_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.24.mlp.up_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.24.mlp.up_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.24.mlp.down_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.24.mlp.down_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.25.self_attn.q_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.25.self_attn.q_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.25.self_attn.k_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.25.self_attn.k_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.25.self_attn.v_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.25.self_attn.v_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.25.self_attn.o_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.25.self_attn.o_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.25.mlp.gate_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.25.mlp.gate_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.25.mlp.up_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.25.mlp.up_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.25.mlp.down_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.25.mlp.down_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.26.self_attn.q_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.26.self_attn.q_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.26.self_attn.k_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.26.self_attn.k_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.26.self_attn.v_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.26.self_attn.v_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.26.self_attn.o_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.26.self_attn.o_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.26.mlp.gate_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.26.mlp.gate_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.26.mlp.up_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.26.mlp.up_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.26.mlp.down_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.26.mlp.down_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.27.self_attn.q_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.27.self_attn.q_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.27.self_attn.k_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.27.self_attn.k_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.27.self_attn.v_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.27.self_attn.v_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.27.self_attn.o_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.27.self_attn.o_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.27.mlp.gate_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.27.mlp.gate_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.27.mlp.up_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.27.mlp.up_proj.lora_B.default.weight', 'base_model.model.model.language_model.layers.27.mlp.down_proj.lora_A.default.weight', 'base_model.model.model.language_model.layers.27.mlp.down_proj.lora_B.default.weight', 'base_model.model.model.prediction_head.layers.0.ffn.gate_proj.lora_A.default.weight', 'base_model.model.model.prediction_head.layers.0.ffn.gate_proj.lora_B.default.weight', 'base_model.model.model.prediction_head.layers.0.ffn.up_proj.lora_A.default.weight', 'base_model.model.model.prediction_head.layers.0.ffn.up_proj.lora_B.default.weight', 'base_model.model.model.prediction_head.layers.0.ffn.down_proj.lora_A.default.weight', 'base_model.model.model.prediction_head.layers.0.ffn.down_proj.lora_B.default.weight', 'base_model.model.model.prediction_head.layers.1.ffn.gate_proj.lora_A.default.weight', 'base_model.model.model.prediction_head.layers.1.ffn.gate_proj.lora_B.default.weight', 'base_model.model.model.prediction_head.layers.1.ffn.up_proj.lora_A.default.weight', 'base_model.model.model.prediction_head.layers.1.ffn.up_proj.lora_B.default.weight', 'base_model.model.model.prediction_head.layers.1.ffn.down_proj.lora_A.default.weight', 'base_model.model.model.prediction_head.layers.1.ffn.down_proj.lora_B.default.weight', 'base_model.model.model.prediction_head.layers.2.ffn.gate_proj.lora_A.default.weight', 'base_model.model.model.prediction_head.layers.2.ffn.gate_proj.lora_B.default.weight', 'base_model.model.model.prediction_head.layers.2.ffn.up_proj.lora_A.default.weight', 'base_model.model.model.prediction_head.layers.2.ffn.up_proj.lora_B.default.weight', 'base_model.model.model.prediction_head.layers.2.ffn.down_proj.lora_A.default.weight', 'base_model.model.model.prediction_head.layers.2.ffn.down_proj.lora_B.default.weight', 'base_model.model.model.prediction_head.layers.3.ffn.gate_proj.lora_A.default.weight', 'base_model.model.model.prediction_head.layers.3.ffn.gate_proj.lora_B.default.weight', 'base_model.model.model.prediction_head.layers.3.ffn.up_proj.lora_A.default.weight', 'base_model.model.model.prediction_head.layers.3.ffn.up_proj.lora_B.default.weight', 'base_model.model.model.prediction_head.layers.3.ffn.down_proj.lora_A.default.weight', 'base_model.model.model.prediction_head.layers.3.ffn.down_proj.lora_B.default.weight'].
  warnings.warn(warn_message)
Merging...
Saving merged model...
Done. Saved to /home/ubuntu/vibevoice_merged