update diffusers to 0.9; add SD2

This commit is contained in:
Qing
2022-12-04 13:41:48 +08:00
parent 15fe87e42d
commit 6a0ffdc96e
7 changed files with 41 additions and 30 deletions

View File

@@ -191,6 +191,8 @@ function ModelSettingBlock() {
return renderFCFModelDesc()
case AIModel.SD15:
return undefined
case AIModel.SD2:
return undefined
case AIModel.Mange:
return undefined
case AIModel.CV2:
@@ -234,10 +236,16 @@ function ModelSettingBlock() {
)
case AIModel.SD15:
return renderModelDesc(
'Stable Diffusion',
'Stable Diffusion 1.5',
'https://ommer-lab.com/research/latent-diffusion-models/',
'https://github.com/CompVis/stable-diffusion'
)
case AIModel.SD2:
return renderModelDesc(
'Stable Diffusion 2',
'https://ommer-lab.com/research/latent-diffusion-models/',
'https://github.com/Stability-AI/stablediffusion'
)
case AIModel.Mange:
return renderModelDesc(
'Manga Inpainting',

View File

@@ -10,6 +10,7 @@ export enum AIModel {
MAT = 'mat',
FCF = 'fcf',
SD15 = 'sd1.5',
SD2 = 'sd2',
CV2 = 'cv2',
Mange = 'manga',
}
@@ -294,7 +295,14 @@ const defaultHDSettings: ModelsHDSettings = {
hdStrategyResizeLimit: 768,
hdStrategyCropTrigerSize: 512,
hdStrategyCropMargin: 128,
enabled: true,
enabled: false,
},
[AIModel.SD2]: {
hdStrategy: HDStrategy.ORIGINAL,
hdStrategyResizeLimit: 768,
hdStrategyCropTrigerSize: 512,
hdStrategyCropMargin: 128,
enabled: false,
},
[AIModel.Mange]: {
hdStrategy: HDStrategy.CROP,
@@ -318,6 +326,7 @@ export enum SDSampler {
klms = 'k_lms',
kEuler = 'k_euler',
kEulerA = 'k_euler_a',
dpmPlusPlus = 'dpm++',
}
export enum SDMode {
@@ -422,7 +431,7 @@ export const isSDState = selector({
key: 'isSD',
get: ({ get }) => {
const settings = get(settingState)
return settings.model === AIModel.SD15
return settings.model === AIModel.SD15 || settings.model === AIModel.SD2
},
})