mirror of https://github.com/THUDM/ChatGLM-6B
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
8 lines
217 B
8 lines
217 B
2 years ago
|
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
|
||
|
|
|