RWKV-Runner/finetune/lora/v6/demo/demo-state-merge.sh
2024-05-28 22:35:47 +08:00

8 lines
291 B
Bash
Vendored

base_model='/home/rwkv/JL/model/RWKV-x060-World-3B-v2.1-20240417-ctx4096.pth'
state_checkpoint='/home/rwkv/JL/out_model/state/rwkv-9.pth'
output='/home/rwkv/JL/model/state-0.pth'
python merge/merge_state.py --base_model $base_model \
--state_checkpoint $state_checkpoint \
--output $output