Change some parameters

This commit is contained in:
oobabooga 2023-03-06 16:29:43 -03:00
parent 20bd645f6a
commit 6904a507c6
2 changed files with 3 additions and 3 deletions

View file

@ -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,

View file

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