瀏覽代碼

Change some parameters

oobabooga 2 年之前
父節點
當前提交
6904a507c6
共有 2 個文件被更改,包括 3 次插入3 次删除
  1. 1 1
      modules/RWKV.py
  2. 2 2
      presets/Naive.txt

+ 1 - 1
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,

+ 2 - 2
presets/Naive.txt

@@ -1,3 +1,3 @@
 do_sample=True
-top_p=0.95
-temperature=0.8
+top_p=0.85
+temperature=1