revert change to _model_generate abstract method signature

3149b6ac · Albert Jiang · 1576e99e · 3149b6ac
Commit 3149b6ac authored Jan 13, 2023 by Albert Jiang
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

lm_eval/base.py lm_eval/base.py +1 -1

No files found.
--- a/lm_eval/base.py
+++ b/lm_eval/base.py
@@ -153,7 +153,7 @@ class BaseLM(LM):
        pass
    @abstractmethod
-    def _model_generate(self, context, max_length, eos_token_id, temperature=0.):
+    def _model_generate(self, context, max_length, eos_token_id):
        pass
    @abstractmethod