Fix max_tokens type in main.py

This commit is contained in:
Sebastián Estévez
2024-04-02 00:28:08 -04:00
committed by GitHub
parent 71db88115d
commit d3a86e7b7f
+2 -2
View File
@@ -179,7 +179,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,
@@ -422,7 +422,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,