diff --git a/ptuning/api.sh b/ptuning/api.sh
new file mode 100644
index 0000000..2fc1663
--- /dev/null
+++ b/ptuning/api.sh
@@ -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