Merge pull request #2790 from phact/patch-2

Fix max_tokens type in main.py
This commit is contained in:
Krish Dholakia
2024-04-02 09:02:34 -07:00
committed by GitHub
+2 -2
View File
@@ -180,7 +180,7 @@ async def acompletion(
n: Optional[int] = None,
stream: Optional[bool] = None,
stop=None,
max_tokens: Optional[float] = None,
max_tokens: Optional[int] = None,
presence_penalty: Optional[float] = None,
frequency_penalty: Optional[float] = None,
logit_bias: Optional[dict] = None,
@@ -423,7 +423,7 @@ def completion(
n: Optional[int] = None,
stream: Optional[bool] = None,
stop=None,
max_tokens: Optional[float] = None,
max_tokens: Optional[int] = None,
presence_penalty: Optional[float] = None,
frequency_penalty: Optional[float] = None,
logit_bias: Optional[dict] = None,