github
/
ChatGLM-6B
mirror of https://github.com/THUDM/ChatGLM-6B
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Create api.sh

pull/928/head
linteng 2023-05-05 15:10:18 +08:00 committed by GitHub
parent 2e8c709251
commit 091f0a4c4d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
ptuning/api.sh Normal file
View File

@ -0,0 +1,6 @@
PRE_SEQ_LEN=128
CUDA_VISIBLE_DEVICES=0 python3 api.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