diff --git a/nodes.py b/nodes.py index 0fb5a4c..8530b28 100644 --- a/nodes.py +++ b/nodes.py @@ -17,18 +17,9 @@ class Generator: "tokens": ("INT", {"default": 128, "min": 1, "max": 8192}), "temp": ("FLOAT", {"default": 0.7, "min": 0.0, "max": 2.0, "step": 0.01}), "top_k": ("INT", {"default": 20, "min": 0, "max": 200}), - "top_p": ( - "FLOAT", - {"default": 0.9, "min": 0.0, "max": 1.0, "step": 0.01}, - ), - "typical_p": ( - "FLOAT", - {"default": 1.0, "min": 0.0, "max": 1.0, "step": 0.01}, - ), - "penalty": ( - "FLOAT", - {"default": 1.15, "min": 1.0, "max": 2.0, "step": 0.01}, - ), + "top_p": ("FLOAT", {"default": 0.9, "min": 0.0, "max": 1.0, "step": 0.01}), + "typical_p": ("FLOAT", {"default": 1.0, "min": 0.0, "max": 1.0, "step": 0.01}), + "penalty": ("FLOAT", {"default": 1.15, "min": 1.0, "max": 2.0, "step": 0.01}), "seed": ("INT", {"default": 0, "min": 0, "max": 2**64 - 1}), "prompt": ("STRING", {"default": "", "multiline": True}), },