From af2c080154ed010916e89ec0218c8afdbbf83bfe Mon Sep 17 00:00:00 2001 From: tluyben Date: Thu, 12 Jan 2023 12:35:52 +0100 Subject: [PATCH] Float -> Int for Top-k Top-k should be a positive integer: ``` Error: top_k has to be a strictly positive integer, but is 3.0 ``` which cannot be fixed because it's a float in web.py. --- web.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/web.py b/web.py index 369b86b..7de33c8 100644 --- a/web.py +++ b/web.py @@ -7,7 +7,7 @@ class Input(BaseModel): generate_tokens_limit: int = 100 top_p: float = 0.7 - top_k: float = 0 + top_k: int = 0 temperature: float = 1.0 text: str