admin777/LaiTool
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Laitool v3.0.1-preview.1

Browse Source
This commit is contained in:
admin777
2024-08-03 12:46:12 +08:00
parent c8a46d59fb
commit 7312053a20
196 changed files with 19822 additions and 8733 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/api/sdApi.js
+72 -76
View File
@@ -1,91 +1,87 @@
import { basicApi } from "./apiBasic";
import { define } from "../define/define";
import { basicApi } from './apiBasic'
import { define } from '../define/define'
import { promises as fspromises } from 'fs'
import { errorMessage, successMessage } from "../main/generalTools";
export class SdApi {
constructor() {
this.baseUrl = global.config?.webui_api_url;
this.sd_setting = null;
constructor() {
this.baseUrl = global.config?.webui_api_url
this.sd_setting = null
}
/**
* 获取当前SD的服务器中所有的lora信息
* @returns
*/
async getAllLoras(baseURL = null) {
let url = this.baseUrl + 'sdapi/v1/loras'
if (baseURL != null) {
url = baseURL + 'sdapi/v1/loras'
}
return await basicApi.get(url)
}
/**
* 获取当前SD的服务器中所有的lora信息
* @returns
*/
async getAllLoras(baseURL = null) {
let url = this.baseUrl + "sdapi/v1/loras";
if (baseURL != null) {
url = baseURL + "sdapi/v1/loras";
}
return await basicApi.get(url);
/**
* 获取当前的所有的checkpoint模型
* @param {*} baseURL
*/
async getAllSDModel(baseURL = null) {
let url = this.baseUrl + 'sdapi/v1/sd-models'
if (baseURL != null) {
url = baseURL + 'sdapi/v1/sd-models'
}
return await basicApi.get(url)
}
/**
* 获取当前的所有的checkpoint模型
* @param {*} baseURL
*/
async getAllSDModel(baseURL = null) {
let url = this.baseUrl + "sdapi/v1/sd-models";
if (baseURL != null) {
url = baseURL + "sdapi/v1/sd-models";
}
return await basicApi.get(url);
/**
* 获取当前连接的所有的samplers（采样器）
* @param {*} baseURL
* @returns
*/
async getAllSamplers(baseURL = null) {
try {
let url = this.baseUrl + 'sdapi/v1/samplers'
if (baseURL != null) {
url = baseURL + 'sdapi/v1/samplers'
}
return await basicApi.get(url)
} catch (error) {
throw error
}
}
/**
* 获取当前连接的所有的samplers（采样器）
* @param {*} baseURL
* @returns
*/
async getAllSamplers(baseURL = null) {
try {
async txt2img(data, baseURL = null) {
try {
if (this.sd_setting == null) {
this.sd_setting = JSON.parse(await fspromises.readFile(define.sd_setting, 'utf-8'))
this.baseUrl = this.sd_setting.setting.webui_api_url
}
let url = this.baseUrl + "sdapi/v1/samplers";
if (baseURL != null) {
url = baseURL + "sdapi/v1/samplers";
}
return await basicApi.get(url);
} catch (error) {
throw error;
}
}
// 加上通用前缀
data.prompt = this.sd_setting.webui.prompt + data.prompt
async txt2img(data, baseURL = null) {
try {
data.negative_prompt = this.sd_setting.webui.negative_prompt
data.sampler_name = this.sd_setting.webui.sampler_name
data.cfg_scale = this.sd_setting.webui.cfg_scale
data.n_iter = 1
data.steps = this.sd_setting.webui.steps
data.save_images = false
data.batch_size = data.batch_size ? data.batch_size : 1
if (this.sd_setting == null) {
this.sd_setting = JSON.parse(await fspromises.readFile(define.sd_setting, 'utf-8'));
this.baseUrl = this.sd_setting.setting.webui_api_url;
}
if (data.width == null) {
data.width = 512
}
if (data.height == null) {
data.height = 512
}
// 加上通用前缀
data.prompt = this.sd_setting.webui.prompt + data.prompt
data.negative_prompt = this.sd_setting.webui.negative_prompt;
data.sampler_name = this.sd_setting.webui.sampler_name;
data.cfg_scale = this.sd_setting.webui.cfg_scale;
data.n_iter = 1;
data.steps = this.sd_setting.webui.steps;
data.save_images = false;
data.batch_size = data.batch_size ? data.batch_size : 1;
if (data.width == null) {
data.width = 512;
}
if (data.height == null) {
data.height = 512;
}
let url = this.baseUrl + "sdapi/v1/txt2img";
if (baseURL != null) {
url = baseURL + "sdapi/v1/txt2img";
}
let res = await basicApi.post(url, data);
return res;
} catch (error) {
throw error;
}
let url = this.baseUrl + 'sdapi/v1/txt2img'
if (baseURL != null) {
url = baseURL + 'sdapi/v1/txt2img'
}
let res = await basicApi.post(url, data)
return res
} catch (error) {
throw error
}
}
}