Merge pull request #929 from EleutherAI/fix-generate-until

Fix `generate_until` rename

Merge pull request #929 from EleutherAI/fix-generate-until
Fix `generate_until` rename
4aafe026 · Lintang Sutawika · GitHub · f8342178 · 785153f6 · 4aafe026
Unverified Commit 4aafe026 authored Oct 18, 2023 by Lintang Sutawika Committed by GitHub Oct 18, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

lm_eval/api/model.py lm_eval/api/model.py +1 -1

No files found.
--- a/lm_eval/api/model.py
+++ b/lm_eval/api/model.py
@@ -96,7 +96,7 @@ class LM(abc.ABC):

    # TODO: Add an optional max length
    @abc.abstractmethod
-    def greedy_until(self, requests) -> List[str]:
+    def generate_until(self, requests) -> List[str]:
        """Generate greedily until a stopping sequence

        :param requests: list[Instance]