PRE_SEQ_LEN=128
LR=1e-4
STEPS=600

CUDA_VISIBLE_DEVICES=0 python3 web_demo.py \
    --model_name_or_path THUDM/chatglm2-6b \
    --ptuning_checkpoint ./ \
    --pre_seq_len $PRE_SEQ_LEN