fix typing (int -> float) (#308)

This commit is contained in:
Harrison Chase 2022-12-10 20:31:55 -08:00 committed by GitHub
parent 9ee6115deb
commit 7827f0a844
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -30,11 +30,11 @@ class OpenAI(LLM, BaseModel):
"""What sampling temperature to use."""
max_tokens: int = 256
"""The maximum number of tokens to generate in the completion."""
top_p: int = 1
top_p: float = 1
"""Total probability mass of tokens to consider at each step."""
frequency_penalty: int = 0
frequency_penalty: float = 0
"""Penalizes repeated tokens according to frequency."""
presence_penalty: int = 0
presence_penalty: float = 0
"""Penalizes repeated tokens."""
n: int = 1
"""How many completions to generate for each prompt."""