From c8d3ed91043c56d671c72016517055de773ec33a Mon Sep 17 00:00:00 2001
From: Maximilian Schmidt <maximilian.schmidt@ims.uni-stuttgart.de>
Date: Fri, 26 Jul 2024 15:35:09 +0200
Subject: [PATCH] Make use of unique prompts for paraphrasing
---
evoprompt/optimization.py | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/evoprompt/optimization.py b/evoprompt/optimization.py
index 2912faa..59b564a 100644
--- a/evoprompt/optimization.py
+++ b/evoprompt/optimization.py
@@ -150,7 +150,10 @@ class PromptOptimization:
self.save_snapshot()
paraphrases, paraphrase_usage = paraphrase_prompts(
- self.evolution_model, self.task.base_prompt, n=num_initial_prompts - 1
+ self.evolution_model,
+ self.task.base_prompt,
+ n=num_initial_prompts - 1,
+ unique_paraphrases=True,
)
self.total_evolution_usage += paraphrase_usage
logger.info("Paraphrased prompt '%s': %s.", self.task.base_prompt.replace("\r", "\\r").replace("\n", "\\n"), paraphrases)
--
GitLab