Commit 4be48911 authored by Zdenek Kasner's avatar Zdenek Kasner
Browse files

e2e_nlg_cleaned: Disable stopping criteria

parent 6588bc74
......@@ -58,8 +58,11 @@ class E2E_NLG_Cleaned(PromptSourceTask):
if self.has_test_docs():
return self.dataset["test"]
def stopping_criteria(self):
return '\n\n'
def max_generation_length(self):
return 64
# def stopping_criteria(self):
# return '\n\n'
def invalid_doc_for_prompt(self, doc) -> bool:
"""The QA prompts are not applicable to all the examples, we want to filter these out."""
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment