Skip to content

GitLab

Commit 3fa62d46 authored by jon-tow's avatar jon-tow
Browse files

Update `max_length`

parent 5bc61283
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
lm_eval/models/gpt3.py
View file @ 3fa62d46
...@@ -72,7 +72,7 @@ def oa_completion(**kwargs): ...@@ -72,7 +72,7 @@ def oa_completion(**kwargs):
class GPT3LM(BaseLM): class GPT3LM(BaseLM):
REQ_CHUNK_SIZE = 40 REQ_CHUNK_SIZE = 20
def __init__(self, engine, truncate=False, api_key=None, pass_strings=False): def __init__(self, engine, truncate=False, api_key=None, pass_strings=False):
""" """
...@@ -276,7 +276,7 @@ class GooseAILM(GPT3LM): ...@@ -276,7 +276,7 @@ class GooseAILM(GPT3LM):
@property @property
def max_length(self): def max_length(self):
# Note: this is temporary, will be raised to 2048 in the future # Note: this is temporary, will be raised to 2048 in the future
return 1023 return 2022
@property @property
def eot_token_id(self): def eot_token_id(self):
......
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