diff --git a/modules/RWKV.py b/modules/RWKV.py index 93b11678..9a806a00 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.7, alpha_presence=0.7, token_ban=[0], token_stop=[], callback=None): + def generate(self, context, token_count=20, temperature=1, top_p=1, alpha_frequency=0.1, alpha_presence=0.1, token_ban=[0], token_stop=[], callback=None): args = PIPELINE_ARGS( temperature = temperature, top_p = top_p, diff --git a/presets/Naive.txt b/presets/Naive.txt index c6965983..f3114a50 100644 --- a/presets/Naive.txt +++ b/presets/Naive.txt @@ -1,3 +1,3 @@ do_sample=True -top_p=0.95 -temperature=0.8 +top_p=0.85 +temperature=1