let5see/IOPaint
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
Qing
2023-03-01 21:44:02 +08:00
parent 8e5e4892af
commit 66d3c6e322
7 changed files with 644 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
lama_cleaner/app/src/components/Settings/ModelSettingBlock.tsx
View File

@@ -230,6 +230,18 @@ function ModelSettingBlock() {
'https://ommer-lab.com/research/latent-diffusion-models/',
'https://github.com/CompVis/stable-diffusion'
)
case AIModel.ANYTHING4:
return renderModelDesc(
'andite/anything-v4.0',
'https://huggingface.co/andite/anything-v4.0',
'https://huggingface.co/andite/anything-v4.0'
)
case AIModel.REALISTIC_VISION_1_4:
return renderModelDesc(
'SG161222/Realistic_Vision_V1.4',
'https://huggingface.co/SG161222/Realistic_Vision_V1.4',
'https://huggingface.co/SG161222/Realistic_Vision_V1.4'
)
case AIModel.SD2:
return renderModelDesc(
'Stable Diffusion 2',

23
lama_cleaner/app/src/store/Atoms.tsx
View File

@@ -10,6 +10,8 @@ export enum AIModel {
MAT = 'mat',
FCF = 'fcf',
SD15 = 'sd1.5',
ANYTHING4 = 'anything4',
REALISTIC_VISION_1_4 = 'realisticVision1.4',
SD2 = 'sd2',
CV2 = 'cv2',
Mange = 'manga',
@@ -422,6 +424,20 @@ const defaultHDSettings: ModelsHDSettings = {
hdStrategyCropMargin: 128,
enabled: false,
},
[AIModel.ANYTHING4]: {
hdStrategy: HDStrategy.ORIGINAL,
hdStrategyResizeLimit: 768,
hdStrategyCropTrigerSize: 512,
hdStrategyCropMargin: 128,
enabled: false,
},
[AIModel.REALISTIC_VISION_1_4]: {
hdStrategy: HDStrategy.ORIGINAL,
hdStrategyResizeLimit: 768,
hdStrategyCropTrigerSize: 512,
hdStrategyCropMargin: 128,
enabled: false,
},
[AIModel.SD2]: {
hdStrategy: HDStrategy.ORIGINAL,
hdStrategyResizeLimit: 768,
@@ -601,7 +617,12 @@ export const isSDState = selector({
key: 'isSD',
get: ({ get }) => {
const settings = get(settingState)
return settings.model === AIModel.SD15 || settings.model === AIModel.SD2
return (
settings.model === AIModel.SD15 ||
settings.model === AIModel.SD2 ||
settings.model === AIModel.ANYTHING4 ||
settings.model === AIModel.REALISTIC_VISION_1_4
)
},
})