Commit 4e764ca0 authored by haileyschoelkopf's avatar haileyschoelkopf
Browse files

Merge branch 'big-refactor' of...

Merge branch 'big-refactor' of https://github.com/EleutherAI/lm-evaluation-harness into big-refactor
parents b7a1214e 6ae376e3
...@@ -15,7 +15,7 @@ from accelerate import Accelerator ...@@ -15,7 +15,7 @@ from accelerate import Accelerator
from itertools import islice from itertools import islice
@register_model("hf-causal", "gpt2") @register_model("hf-causal")
class HFLM(LM): class HFLM(LM):
def __init__( def __init__(
self, self,
......
...@@ -57,7 +57,7 @@ def oa_completion(**kwargs): ...@@ -57,7 +57,7 @@ def oa_completion(**kwargs):
backoff_time *= 1.5 backoff_time *= 1.5
@register_model("openai") @register_model("openai"., "gooseai")
class GPT3LM(LM): class GPT3LM(LM):
REQ_CHUNK_SIZE = 20 REQ_CHUNK_SIZE = 20
......
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