add nai diffusion

Signed-off-by: hashcoko <hashcoko@gmail.com>
This commit is contained in:
hashcoko
2023-11-27 07:13:22 +09:00
parent d9e16cc427
commit b69f31caae
12 changed files with 226 additions and 40 deletions

View File

@@ -175,7 +175,7 @@ export function setDatabase(data:Database){
data.sdCFG = 7
}
if(checkNullish(data.NAIImgUrl)){
data.NAIImgUrl = 'https://api.novelai.net'
data.NAIImgUrl = 'https://api.novelai.net/ai/generate-image'
}
if(checkNullish(data.NAIApiKey)){
data.NAIApiKey = ''
@@ -183,6 +183,9 @@ export function setDatabase(data:Database){
if(checkNullish(data.NAIImgModel)){
data.NAIImgModel = 'nai-diffusion-3'
}
if(checkNullish(data.NAII2I)){
data.NAII2I = true
}
if(checkNullish(data.textTheme)){
data.textTheme = "standard"
}
@@ -248,7 +251,9 @@ export function setDatabase(data:Database){
steps:28,
scale:5,
sm:true,
sm_dyn:true
sm_dyn:true,
noise:0.0,
strength:0.3
}
}
if(checkNullish(data.customTextTheme)){
@@ -417,6 +422,7 @@ export interface Database{
NAIImgUrl:string
NAIApiKey:string
NAIImgModel:string
NAII2I:boolean
NAIImgConfig:NAIImgConfig
runpodKey:string
promptPreprocess:boolean
@@ -748,7 +754,9 @@ interface NAIImgConfig{
steps:number,
scale:number,
sm:boolean,
sm_dyn:boolean
sm_dyn:boolean,
noise:number,
strength:number
}
export type FormatingOrderItem = 'main'|'jailbreak'|'chats'|'lorebook'|'globalNote'|'authorNote'|'lastChat'|'description'|'postEverything'|'personaPrompt'

View File

@@ -661,7 +661,7 @@ export async function globalFetch(url:string, arg:{
method: method,
signal: arg.abortSignal
})
addFetchLog("Uint8Array Response", da.ok && da.status >= 200 && da.status < 300)
return {
ok: da.ok && da.status >= 200 && da.status < 300,
@@ -685,6 +685,16 @@ export async function globalFetch(url:string, arg:{
headers: headers,
method: method
})
if(da.headers.get('content-type')?.includes('application/x-zip-compressed')){
const daText = await da.blob()
addFetchLog(daText, da.ok && da.status >= 200 && da.status < 300)
return {
ok: da.ok && da.status >= 200 && da.status < 300,
data: daText,
headers: Object.fromEntries(da.headers)
}
}
const daText = await da.text()
try {
const dat = JSON.parse(daText)