Skip to content

GitLab

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

Remove custom max_length method

parent 5a62765b
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 4 deletions
lm_eval/tasks/e2e_nlg_cleaned.py
View file @ 9aba053a
...@@ -61,10 +61,6 @@ class E2E_NLG_Cleaned(PromptSourceTask): ...@@ -61,10 +61,6 @@ class E2E_NLG_Cleaned(PromptSourceTask):
def stopping_criteria(self): def stopping_criteria(self):
return '\n' return '\n'
def max_generation_length(self):
# TODO check
return 512
def invalid_doc_for_prompt(self, doc) -> bool: def invalid_doc_for_prompt(self, doc) -> bool:
"""The QA prompts are not applicable to all the examples, we want to filter these out.""" """The QA prompts are not applicable to all the examples, we want to filter these out."""
return self.prompt.name.endswith("_qa") 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