File size: 187 Bytes
c0f9977 |
1 2 3 4 5 6 7 8 9 10 |
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
|