Commit 9aba053a authored by Zdenek Kasner's avatar Zdenek Kasner
Browse files

Remove custom max_length method

parent 5a62765b
......@@ -61,10 +61,6 @@ class E2E_NLG_Cleaned(PromptSourceTask):
def stopping_criteria(self):
return '\n'
def max_generation_length(self):
# TODO check
return 512
def invalid_doc_for_prompt(self, doc) -> bool:
"""The QA prompts are not applicable to all the examples, we want to filter these out."""
return self.prompt.name.endswith("_qa")
......
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