Skip to content

GitLab

Unverified Commit 25a15379 authored by Stella Biderman's avatar Stella Biderman Committed by GitHub
Browse files

Update openai_completions.py (#1238)

parent 4d10ad56
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
lm_eval/models/openai_completions.py
View file @ 25a15379
...@@ -76,7 +76,7 @@ class OpenaiCompletionsLM(LM): ...@@ -76,7 +76,7 @@ class OpenaiCompletionsLM(LM):
def __init__( def __init__(
self, self,
model: str = "text-davinci-003", model: str,
truncate: bool = False, truncate: bool = False,
max_gen_toks: int = 256, max_gen_toks: int = 256,
batch_size: int = 1, batch_size: int = 1,
...@@ -86,7 +86,7 @@ class OpenaiCompletionsLM(LM): ...@@ -86,7 +86,7 @@ class OpenaiCompletionsLM(LM):
""" """
:param engine: str :param engine: str
OpenAI API engine (e.g. davinci) OpenAI API engine (e.g. gpt-3.5-turbo-instruct)
:param truncate: bool :param truncate: bool
Truncate input if too long (if False and input is too long, throw error) Truncate input if too long (if False and input is too long, throw error)
""" """
......
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