From 6904a507c661c9d941d6ff643b16589c6f670905 Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Mon, 6 Mar 2023 16:29:43 -0300
Subject: [PATCH] Change some parameters

---
 modules/RWKV.py   | 2 +-
 presets/Naive.txt | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

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