From 5bed607b773886b0c5ade204f63f0e5cdb3e502e Mon Sep 17 00:00:00 2001 From: oobabooga <112222186+oobabooga@users.noreply.github.com> Date: Mon, 6 Mar 2023 14:25:48 -0300 Subject: [PATCH] Increase repetition frequency/penalty for RWKV --- modules/RWKV.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/RWKV.py b/modules/RWKV.py index 2ef86cae..93b11678 100644 --- a/modules/RWKV.py +++ b/modules/RWKV.py @@ -33,7 +33,7 @@ class RWKVModel: result.pipeline = pipeline return result - def generate(self, context, token_count=20, temperature=1, top_p=1, alpha_frequency=0.25, alpha_presence=0.25, token_ban=[0], token_stop=[], callback=None): + def generate(self, context, token_count=20, temperature=1, top_p=1, alpha_frequency=0.7, alpha_presence=0.7, token_ban=[0], token_stop=[], callback=None): args = PIPELINE_ARGS( temperature = temperature, top_p = top_p,