Fix GUI Kobold

This commit is contained in:
Cohee 2024-01-02 10:28:34 +02:00
parent 9b24e7dc67
commit c69724e1da
2 changed files with 2 additions and 3 deletions

View File

@ -2891,7 +2891,7 @@ export async function generateRaw(prompt, api, instructOverride) {
case 'kobold': case 'kobold':
case 'koboldhorde': case 'koboldhorde':
if (preset_settings === 'gui') { if (preset_settings === 'gui') {
generateData = { prompt: prompt, gui_settings: true, max_length: amount_gen, max_context_length: max_context }; generateData = { prompt: prompt, gui_settings: true, max_length: amount_gen, max_context_length: max_context, api_server };
} else { } else {
const isHorde = api === 'koboldhorde'; const isHorde = api === 'koboldhorde';
const koboldSettings = koboldai_settings[koboldai_setting_names[preset_settings]]; const koboldSettings = koboldai_settings[koboldai_setting_names[preset_settings]];
@ -3724,6 +3724,7 @@ async function Generate(type, { automatic_trigger, force_name2, quiet_prompt, qu
gui_settings: true, gui_settings: true,
max_length: maxLength, max_length: maxLength,
max_context_length: max_context, max_context_length: max_context,
api_server,
}; };
if (preset_settings != 'gui') { if (preset_settings != 'gui') {

View File

@ -4,7 +4,6 @@ import {
getStoppingStrings, getStoppingStrings,
substituteParams, substituteParams,
api_server, api_server,
main_api,
} from '../script.js'; } from '../script.js';
import { import {
@ -142,7 +141,6 @@ export function getKoboldGenerationData(finalPrompt, settings, maxLength, maxCon
sampler_seed: kai_settings.seed >= 0 ? kai_settings.seed : undefined, sampler_seed: kai_settings.seed >= 0 ? kai_settings.seed : undefined,
api_server, api_server,
main_api,
}; };
return generate_data; return generate_data;
} }