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 | |
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 | |