Removed circular list infinite loop

curt-tigges · Nov 22, 2023 · 05644d9 · 05644d9
1 parent ee437c6
commit 05644d9
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/utils/prompts.py b/utils/prompts.py
@@ -675,8 +675,10 @@ def get_prompts(
         neg_prompts = [p + " " for p in neg_prompts]
         if neutral_prompts is not None:
             neutral_prompts = [p + " " for p in neutral_prompts]
-        pos_answers = CircularList([p.strip() + " " for p in pos_answers])
-        neg_answers = CircularList([p.strip() + " " for p in neg_answers])
+        for i in range(len(pos_answers)):
+            pos_answers[i] = pos_answers[i].strip()
+        for i in range(len(neg_answers)):
+            neg_answers[i] = neg_answers[i].strip()
 
     # create output dicts
     prompt_dict = dict(