Switch gpt3 to BaseLM

8ef7a515 · Leo Gao · fea936e6 · 8ef7a515
Commit 8ef7a515 authored Nov 05, 2021 by Leo Gao
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

lm_eval/models/gpt3.py lm_eval/models/gpt3.py +2 -2

No files found.
--- a/lm_eval/models/gpt3.py
+++ b/lm_eval/models/gpt3.py
 import os
 import numpy as np
 import transformers
-from lm_eval.base import LM, TokenizedLM
+from lm_eval.base import LM, BaseLM
 from lm_eval import utils
 from tqdm import tqdm
 import time
@@ -35,7 +35,7 @@ def oa_completion(**kwargs):
            backoff_time *= 1.5


-class GPT3LM(TokenizedLM):
+class GPT3LM(BaseLM):
    REQ_CHUNK_SIZE = 20

    def __init__(self, engine, truncate=False):