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
+3 -3
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."""