1
0
Fork 0

send experimental params to server

This commit is contained in:
Sean Sube 2024-02-17 15:31:34 -06:00
parent 2449e347f0
commit 749cf84374
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 29 additions and 2 deletions

View File

@ -17,6 +17,7 @@ import { ExtrasFile } from '../types/model.js';
import { import {
BaseImgParams, BaseImgParams,
BlendParams, BlendParams,
ExperimentalParams,
HighresParams, HighresParams,
ImageSize, ImageSize,
Img2ImgJSONParams, Img2ImgJSONParams,
@ -60,12 +61,21 @@ export interface ImageJSON {
highres?: HighresParams; highres?: HighresParams;
img2img?: Img2ImgJSONParams; img2img?: Img2ImgJSONParams;
inpaint?: InpaintJSONParams; inpaint?: InpaintJSONParams;
experimental?: ExperimentalParams;
}
export interface JSONInner {
[key: string]: string | number | boolean | undefined | JSONInner;
}
export interface JSONBody extends JSONInner {
params: JSONInner;
} }
export function makeImageJSON(params: ImageJSON): string { export function makeImageJSON(params: ImageJSON): string {
const { model, base, img2img, inpaint, size, border, upscale, highres } = params; const { model, base, img2img, inpaint, size, border, upscale, highres, experimental } = params;
const body: Record<string, Record<string, string | number | boolean | undefined>> = { const body: JSONBody = {
device: { device: {
platform: model.platform, platform: model.platform,
}, },
@ -153,6 +163,23 @@ export function makeImageJSON(params: ImageJSON): string {
}; };
} }
if (doesExist(experimental)) {
body.experimental = {
latentSymmetry: {
enabled: experimental.latentSymmetry.enabled,
gradientStart: experimental.latentSymmetry.gradientStart,
gradientEnd: experimental.latentSymmetry.gradientEnd,
lineOfSymmetry: experimental.latentSymmetry.lineOfSymmetry,
},
promptEditing: {
enabled: experimental.promptEditing.enabled,
promptFilter: experimental.promptEditing.filter,
removeTokens: experimental.promptEditing.removeTokens,
addSuffix: experimental.promptEditing.addSuffix,
},
};
}
return JSON.stringify(body); return JSON.stringify(body);
} }