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

Update openai_completions.py (#1238)

parent 4d10ad56
......@@ -76,7 +76,7 @@ class OpenaiCompletionsLM(LM):
def __init__(
self,
model: str = "text-davinci-003",
model: str,
truncate: bool = False,
max_gen_toks: int = 256,
batch_size: int = 1,
......@@ -86,7 +86,7 @@ class OpenaiCompletionsLM(LM):
"""
:param engine: str
OpenAI API engine (e.g. davinci)
OpenAI API engine (e.g. gpt-3.5-turbo-instruct)
:param truncate: bool
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