From 091f0a4c4ddbb6f2325b7c84d4c30c6be9924e45 Mon Sep 17 00:00:00 2001
From: linteng <10499295+wanglinteng@users.noreply.github.com>
Date: Fri, 5 May 2023 15:10:18 +0800
Subject: [PATCH] Create api.sh

---
 ptuning/api.sh | 6 ++++++
 1 file changed, 6 insertions(+)
 create mode 100644 ptuning/api.sh

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