codys12
/

Mistral-7b-Pathway-128k-3

Model card Files Files and versions

codys12 commited on Nov 18, 2023

Commit

428f41a

·

1 Parent(s): 71ee6de

Upload handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -155,7 +155,7 @@ def parse(generated_text: str, conclusions: list[tuple[str, str]], end_sequences
         for conclusion_key, _ in conclusions:
             if conclusion_key in generated_text:
                 conclusion_found = conclusion_key
-                break
     # Extract context numbers from the generated text
     context_pattern = r"\[\d+\]"

         for conclusion_key, _ in conclusions:
             if conclusion_key in generated_text:
                 conclusion_found = conclusion_key
+                generated_text = generated_text.replace(conclusion_key, "")
     # Extract context numbers from the generated text
     context_pattern = r"\[\d+\]"