codys12
/

Mistral-7b-Pathway-128k-3

Model card Files Files and versions

codys12 commited on Nov 18, 2023

Commit

a929d20

·

1 Parent(s): 221fe9f

Upload handler.py

Files changed (1) hide show

handler.py +1 -0

handler.py CHANGED Viewed

@@ -135,6 +135,7 @@ def generate(
             if end_sequence in generated_text:
                 generated_text = generated_text.replace(end_sequence, "")
                 return parse(generated_text, conclusions, end_sequences)
 def parse(generated_text: str, conclusions: list[tuple[str, str]], end_sequences: list[str]) -> dict:
     # Initialize variables

             if end_sequence in generated_text:
                 generated_text = generated_text.replace(end_sequence, "")
                 return parse(generated_text, conclusions, end_sequences)
+    return parse(generated_text, conclusions, end_sequences)
 def parse(generated_text: str, conclusions: list[tuple[str, str]], end_sequences: list[str]) -> dict:
     # Initialize variables