RWKV-PEFT-Web 配置后训练报错

我启动了 RWKV-PEFT 的 Web 服务,并在 Web 端配置了如下参数:

显示的脚本预览:

python train.py \
--load_model /home/rwkv/models/basemodel/1.6b.pth \
--proj_dir /home/rwkv/peft-out-model/bone-test --data_file /home/rwkv/data/lora-test \
--data_type binidx --vocab_size 65536 \
--ctx_len 512 --epoch_steps 100 --epoch_count 1 \
--epoch_begin 0 --epoch_save 1 --micro_bsz 4 \
--n_layer 24 --n_embd 2048 \
--lr_init 2e-05 --lr_final 2e-05 --warmup_steps 0 \
--pre_ffn 0 --head_qk 0 --beta1 0.9 --beta2 0.99 --adam_eps 1e-8 --accelerator gpu \
--devices 1 --precision bf16 --strategy deepspeed_stage_1 --grad_cp 1 \
--dataload pad --loss_mask none --data_shuffle 1 \
--fla \
--my_testing "x060" \
--quant int8 \
--peft bone \
--bone_config '{"bone_load": "", "bone_r": 64}'

点击 running train 按钮后,Web 端显示正在启动训练:

但实际上终端出现 KeyError: 'bone_mode'报错,显示如下:

请问我的配置参数应当如何调整?

等待,gradio 部分修复