* feat: add engine settings Signed-off-by: James <james@jan.ai> --------- Signed-off-by: James <james@jan.ai> Co-authored-by: Louis <louis@jan.ai>
141 lines
2.9 KiB
TypeScript
141 lines
2.9 KiB
TypeScript
import { SettingComponentData } from './settingComponentBuilder'
|
|
|
|
export const presetConfiguration: Record<string, SettingComponentData> = {
|
|
prompt_template: {
|
|
name: 'prompt_template',
|
|
title: 'Prompt template',
|
|
description: 'Prompt template',
|
|
controllerType: 'input',
|
|
controllerData: {
|
|
placeholder: 'Prompt template',
|
|
value: '',
|
|
},
|
|
},
|
|
stop: {
|
|
name: 'stop',
|
|
title: 'Stop',
|
|
description: 'Stop',
|
|
controllerType: 'input',
|
|
controllerData: {
|
|
placeholder: 'Stop',
|
|
value: '',
|
|
},
|
|
},
|
|
ctx_len: {
|
|
name: 'ctx_len',
|
|
title: 'Context Length',
|
|
description: 'Context Length',
|
|
controllerType: 'slider',
|
|
controllerData: {
|
|
min: 0,
|
|
max: 4096,
|
|
step: 128,
|
|
value: 1024,
|
|
},
|
|
},
|
|
max_tokens: {
|
|
name: 'max_tokens',
|
|
title: 'Max Tokens',
|
|
description:
|
|
'The maximum number of tokens the model will generate in a single response.',
|
|
controllerType: 'slider',
|
|
controllerData: {
|
|
min: 0,
|
|
max: 4096,
|
|
step: 128,
|
|
value: 2048,
|
|
},
|
|
},
|
|
ngl: {
|
|
name: 'ngl',
|
|
title: 'NGL',
|
|
description: 'Number of layers in the neural network.',
|
|
controllerType: 'slider',
|
|
controllerData: {
|
|
min: 1,
|
|
max: 100,
|
|
step: 1,
|
|
value: 100,
|
|
},
|
|
},
|
|
embedding: {
|
|
name: 'embedding',
|
|
title: 'Embedding',
|
|
description: 'Indicates if embedding layers are used.',
|
|
controllerType: 'checkbox',
|
|
controllerData: {
|
|
checked: true,
|
|
},
|
|
},
|
|
stream: {
|
|
name: 'stream',
|
|
title: 'Stream',
|
|
description: 'Stream',
|
|
controllerType: 'checkbox',
|
|
controllerData: {
|
|
checked: false,
|
|
},
|
|
},
|
|
temperature: {
|
|
name: 'temperature',
|
|
title: 'Temperature',
|
|
description:
|
|
"Controls randomness in model's responses. Higher values lead to more random responses.",
|
|
controllerType: 'slider',
|
|
controllerData: {
|
|
min: 0,
|
|
max: 2,
|
|
step: 0.1,
|
|
value: 0.7,
|
|
},
|
|
},
|
|
frequency_penalty: {
|
|
name: 'frequency_penalty',
|
|
title: 'Frequency Penalty',
|
|
description: 'Frequency Penalty',
|
|
controllerType: 'slider',
|
|
controllerData: {
|
|
min: 0,
|
|
max: 1,
|
|
step: 0.1,
|
|
value: 0.7,
|
|
},
|
|
},
|
|
presence_penalty: {
|
|
name: 'presence_penalty',
|
|
title: 'Presence Penalty',
|
|
description: 'Presence Penalty',
|
|
controllerType: 'slider',
|
|
controllerData: {
|
|
min: 0,
|
|
max: 1,
|
|
step: 0.1,
|
|
value: 0.7,
|
|
},
|
|
},
|
|
top_p: {
|
|
name: 'top_p',
|
|
title: 'Top P',
|
|
description: 'Top P',
|
|
controllerType: 'slider',
|
|
controllerData: {
|
|
min: 0,
|
|
max: 1,
|
|
step: 0.1,
|
|
value: 0.95,
|
|
},
|
|
},
|
|
n_parallel: {
|
|
name: 'n_parallel',
|
|
title: 'N Parallel',
|
|
description: 'N Parallel',
|
|
controllerType: 'slider',
|
|
controllerData: {
|
|
min: 1,
|
|
max: 4,
|
|
step: 1,
|
|
value: 1,
|
|
},
|
|
},
|
|
}
|