diff --git a/conversers.py b/conversers.py index f7c76b0..93a7736 100644 --- a/conversers.py +++ b/conversers.py @@ -103,9 +103,9 @@ def get_defense_prompt(self, conv, prompt): init_message = """{\"improvement\": \"""" full_prompts = [] - task_prompt = f'''Please optimize the defense prompt based on the response of vision-language moodel to a jailbreak instruction from scenario {self.scenario}. \n + task_prompt = f'''Please optimize the defense prompt based on the response of vision-language model to a jailbreak instruction from scenario {self.scenario}. \n The safety rules of scenario {self.scenario} is {self.rule} - You should optimize defense prompt based on the specific rules to improve the vision-language model's safety. \n The response of vision-language moodel is: "{prompt}" ''' + You should optimize defense prompt based on the specific rules to improve the vision-language model's safety. \n The response of vision-language model is: "{prompt}" ''' conv.append_message(conv.roles[0], task_prompt) @@ -411,4 +411,4 @@ def get_model_path_and_template(model_name): - \ No newline at end of file +