mirror of
https://github.com/oobabooga/text-generation-webui.git
synced 2024-11-25 17:29:22 +01:00
Change some parameters
This commit is contained in:
parent
20bd645f6a
commit
6904a507c6
@ -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,
|
||||
|
@ -1,3 +1,3 @@
|
||||
do_sample=True
|
||||
top_p=0.95
|
||||
temperature=0.8
|
||||
top_p=0.85
|
||||
temperature=1
|
||||
|
Loading…
Reference in New Issue
Block a user