| PRE_SEQ_LEN=128 | |
| CUDA_VISIBLE_DEVICES=0 python3 web_demo.py \ | |
| --model_name_or_path THUDM/chatglm-6b \ | |
| --ptuning_checkpoint output/adgen-chatglm-6b-pt-128-2e-2/checkpoint-3000 \ | |
| --pre_seq_len $PRE_SEQ_LEN | |
| PRE_SEQ_LEN=128 | |
| CUDA_VISIBLE_DEVICES=0 python3 web_demo.py \ | |
| --model_name_or_path THUDM/chatglm-6b \ | |
| --ptuning_checkpoint output/adgen-chatglm-6b-pt-128-2e-2/checkpoint-3000 \ | |
| --pre_seq_len $PRE_SEQ_LEN | |