Add shared defaults for AI parameters
This commit is contained in:
parent
0c7d40a56d
commit
32e79a13d5
3 changed files with 19 additions and 26 deletions
|
@ -1,5 +1,6 @@
|
||||||
"use strict";
|
"use strict";
|
||||||
import { aiLog, setDebug } from "../logger.js";
|
import { aiLog, setDebug } from "../logger.js";
|
||||||
|
import { DEFAULT_AI_PARAMS } from "./defaultParams.js";
|
||||||
|
|
||||||
const storage = (globalThis.messenger ?? globalThis.browser).storage;
|
const storage = (globalThis.messenger ?? globalThis.browser).storage;
|
||||||
|
|
||||||
|
@ -33,19 +34,7 @@ let gCustomTemplate = "";
|
||||||
let gCustomSystemPrompt = DEFAULT_CUSTOM_SYSTEM_PROMPT;
|
let gCustomSystemPrompt = DEFAULT_CUSTOM_SYSTEM_PROMPT;
|
||||||
let gTemplateText = "";
|
let gTemplateText = "";
|
||||||
|
|
||||||
let gAiParams = {
|
let gAiParams = Object.assign({}, DEFAULT_AI_PARAMS);
|
||||||
max_tokens: 4096,
|
|
||||||
temperature: 0.6,
|
|
||||||
top_p: 0.95,
|
|
||||||
seed: -1,
|
|
||||||
repetition_penalty: 1.0,
|
|
||||||
top_k: 20,
|
|
||||||
min_p: 0,
|
|
||||||
presence_penalty: 0,
|
|
||||||
frequency_penalty: 0,
|
|
||||||
typical_p: 1,
|
|
||||||
tfs: 1,
|
|
||||||
};
|
|
||||||
|
|
||||||
let gCache = new Map();
|
let gCache = new Map();
|
||||||
let gCacheLoaded = false;
|
let gCacheLoaded = false;
|
||||||
|
|
16
modules/defaultParams.js
Normal file
16
modules/defaultParams.js
Normal file
|
@ -0,0 +1,16 @@
|
||||||
|
"use strict";
|
||||||
|
|
||||||
|
export const DEFAULT_AI_PARAMS = {
|
||||||
|
max_tokens: 4096,
|
||||||
|
temperature: 0.6,
|
||||||
|
top_p: 0.95,
|
||||||
|
seed: -1,
|
||||||
|
repetition_penalty: 1.0,
|
||||||
|
top_k: 20,
|
||||||
|
min_p: 0,
|
||||||
|
presence_penalty: 0,
|
||||||
|
frequency_penalty: 0,
|
||||||
|
typical_p: 1,
|
||||||
|
tfs: 1,
|
||||||
|
};
|
||||||
|
|
|
@ -4,6 +4,7 @@ document.addEventListener('DOMContentLoaded', async () => {
|
||||||
const AiClassifier = await import(browser.runtime.getURL('modules/AiClassifier.js'));
|
const AiClassifier = await import(browser.runtime.getURL('modules/AiClassifier.js'));
|
||||||
const dataTransfer = await import(browser.runtime.getURL('options/dataTransfer.js'));
|
const dataTransfer = await import(browser.runtime.getURL('options/dataTransfer.js'));
|
||||||
const { detectSystemTheme } = await import(browser.runtime.getURL('modules/themeUtils.js'));
|
const { detectSystemTheme } = await import(browser.runtime.getURL('modules/themeUtils.js'));
|
||||||
|
const { DEFAULT_AI_PARAMS } = await import(browser.runtime.getURL('modules/defaultParams.js'));
|
||||||
const defaults = await storage.local.get([
|
const defaults = await storage.local.get([
|
||||||
'endpoint',
|
'endpoint',
|
||||||
'templateName',
|
'templateName',
|
||||||
|
@ -66,19 +67,6 @@ document.addEventListener('DOMContentLoaded', async () => {
|
||||||
markDirty();
|
markDirty();
|
||||||
await applyTheme(themeSelect.value);
|
await applyTheme(themeSelect.value);
|
||||||
});
|
});
|
||||||
const DEFAULT_AI_PARAMS = {
|
|
||||||
max_tokens: 4096,
|
|
||||||
temperature: 0.6,
|
|
||||||
top_p: 0.95,
|
|
||||||
seed: -1,
|
|
||||||
repetition_penalty: 1.0,
|
|
||||||
top_k: 20,
|
|
||||||
min_p: 0,
|
|
||||||
presence_penalty: 0,
|
|
||||||
frequency_penalty: 0,
|
|
||||||
typical_p: 1,
|
|
||||||
tfs: 1
|
|
||||||
};
|
|
||||||
document.getElementById('endpoint').value = defaults.endpoint || 'http://127.0.0.1:5000/v1/completions';
|
document.getElementById('endpoint').value = defaults.endpoint || 'http://127.0.0.1:5000/v1/completions';
|
||||||
|
|
||||||
const templates = {
|
const templates = {
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue