From 23b1daa3c69b33fdb3e2e5e763e00bd30ba3201c Mon Sep 17 00:00:00 2001 From: Aleksander Majda Date: Mon, 25 Mar 2024 00:49:32 +0100 Subject: [PATCH] update README, set top_k min to 0 --- LLM_Node.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/LLM_Node.py b/LLM_Node.py index f0fa5f9..560ed84 100644 --- a/LLM_Node.py +++ b/LLM_Node.py @@ -31,7 +31,7 @@ class LLM_Node: "max_tokens": ("INT", {"default": 2000, "min": 1}), "temperature": ("FLOAT", {"default": 1.0, "min": 0.1, "step": 0.1}), "top_p": ("FLOAT", {"default": 0.9, "min": 0.1, "step": 0.1}), - "top_k": ("INT", {"default": 50, "min": 1}), + "top_k": ("INT", {"default": 50, "min": 0}), "repetition_penalty": ("FLOAT", {"default": 1.2, "min": 0.1, "step": 0.1}), "trust_remote_code": ("BOOLEAN", {"default": False}), "torch_dtype": (dtype_options, {"default": "auto"}),