lora error in stage 1

use lora param in stage1
here is context:

deepspeed main.py \
   --lora_dim 8 --only_optimize_lora \
   --data_path wangrui6/Zhihu-KOL Cohere/miracl-zh-queries-22-12 Hello-SimpleAI/HC3-Chinese mkqa-Chinese \
   --data_split 10,0,0 \
   --model_name_or_path $MODEL \
   --per_device_train_batch_size 2 \
   --per_device_eval_batch_size 2 \
   --learning_rate 9.65e-6 \
   --num_train_epochs 16  \
   --deepspeed --seed 1234 --num_warmup_steps 0 \
   --lr_scheduler_type cosine \
   --output_dir $OUTPUT_PATH \
   &> $OUTPUT_PATH/training.log

add about lora
--lora_dim 8 --only_optimize_lora \

get error:

RuntimeError: element 0 of tensors does not require grad and does not have a grad_fn