From ada7c56defb88af80c949185ba3abe8e5feaf9e7 Mon Sep 17 00:00:00 2001
From: redwyd <wangyidan@iie.ac.cn>
Date: Sun, 11 May 2025 22:08:44 +0800
Subject: [PATCH] Init Commit

---
 run.sh | 48 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 48 insertions(+)
 create mode 100644 run.sh

diff --git a/run.sh b/run.sh
new file mode 100644
index 0000000..0df9708
--- /dev/null
+++ b/run.sh
@@ -0,0 +1,48 @@
+#!/usr/bin/env bash
+
+export CUDA_VISIBLE_DEVICES=1
+
+jailbreak_method="GCG"
+
+dataset_name="enron" # "enron", "trustllm"
+data_file="${dataset_name}.jsonl" # "privacy_leakage.jsonl", "privacy_awareness_query.jsonl", "privacy_awareness_confAIde.jsonl"
+data_path=./data/${dataset_name}/${data_file}
+
+target_model_name="llama-2"
+target_model_dir="llama2-7b-chat-hf"
+
+target_model_path=/data/wangyidan/model/${target_model_dir}
+
+attack_model_name="llama-2"
+attack_model_dir="llama2-7b-chat-hf"
+
+attack_model_path=/data/wangyidan/model/${attack_model_dir}
+
+eval_model_name='None'
+eval_model_dir='None'
+
+eval_model_path=/data/wangyidan/model/${eval_model_dir}
+
+output_json_filepath=./output/${jailbreak_method}/${dataset_name}
+output_json_filename=${target_model_dir}.jsonl
+
+mkdir -p ${output_json_filepath}
+
+LOG_DIR=$(pwd)/log/${jailbreak_method}/${dataset_name}
+LOG_FILE=${LOG_DIR}/${target_model_dir}.log
+
+mkdir -p "${LOG_DIR}"
+
+nohup python -u attack.py \
+    --data_path=${data_path} \
+    --dataset_name=${dataset_name} \
+    --attack_model_name=${attack_model_name} \
+    --attack_model_path=${attack_model_path} \
+    --target_model_name=${target_model_name} \
+    --target_model_path=${target_model_path} \
+    --eval_model_name=${eval_model_name} \
+    --eval_model_path=${eval_model_path} \
+    --jailbreak_method=${jailbreak_method} \
+    --output_json_filepath=${output_json_filepath} \
+    --output_json_filename=${output_json_filename} \
+> "$LOG_FILE" 2>&1 &
\ No newline at end of file