From 20c166db4ca4c60a8a0857769d16b5f475ea4fa0 Mon Sep 17 00:00:00 2001
From: p-ferreira <38992619+p-ferreira@users.noreply.github.com>
Date: Wed, 1 Nov 2023 09:51:15 -0400
Subject: [PATCH] lower diversity boundary to 0.2

---
 prompting/validators/reward/diversity.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/prompting/validators/reward/diversity.py b/prompting/validators/reward/diversity.py
index d0a5b77..66314c5 100644
--- a/prompting/validators/reward/diversity.py
+++ b/prompting/validators/reward/diversity.py
@@ -68,7 +68,7 @@ def __init__(self, device: str):
         self.history_reward_bottom_k = 2
         self.historic_embeddings = torch.tensor([]).to(self.device)
         self.history_range = (500, 15500)
-        self.boundary = 0.5
+        self.boundary = 0.2
 
     def get_embeddings(self, sentences: List[str]) -> "torch.FloatTensor":
         """Runs a forward pass through the model.