Allow API requests to use parameter presets
This commit is contained in:
parent
8936160e54
commit
474dc7355a
8 changed files with 96 additions and 58 deletions
|
@ -12,8 +12,12 @@ def run(prompt):
|
|||
request = {
|
||||
'prompt': prompt,
|
||||
'max_new_tokens': 250,
|
||||
|
||||
# Generation params. If 'preset' is set to different than 'None', the values
|
||||
# in presets/preset-name.yaml are used instead of the individual numbers.
|
||||
'preset': 'None',
|
||||
'do_sample': True,
|
||||
'temperature': 1.3,
|
||||
'temperature': 0.7,
|
||||
'top_p': 0.1,
|
||||
'typical_p': 1,
|
||||
'epsilon_cutoff': 0, # In units of 1e-4
|
||||
|
@ -31,6 +35,7 @@ def run(prompt):
|
|||
'mirostat_mode': 0,
|
||||
'mirostat_tau': 5,
|
||||
'mirostat_eta': 0.1,
|
||||
|
||||
'seed': -1,
|
||||
'add_bos_token': True,
|
||||
'truncation_length': 2048,
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue