add MAT model
This commit is contained in:
@@ -131,6 +131,8 @@ function ModelSettingBlock() {
|
||||
return renderLDMModelDesc()
|
||||
case AIModel.ZITS:
|
||||
return renderZITSModelDesc()
|
||||
case AIModel.MAT:
|
||||
return undefined
|
||||
default:
|
||||
return <></>
|
||||
}
|
||||
@@ -156,6 +158,12 @@ function ModelSettingBlock() {
|
||||
'https://arxiv.org/abs/2203.00867',
|
||||
'https://github.com/DQiaole/ZITS_inpainting'
|
||||
)
|
||||
case AIModel.MAT:
|
||||
return renderModelDesc(
|
||||
'Mask-Aware Transformer for Large Hole Image Inpainting',
|
||||
'https://arxiv.org/pdf/2203.15270.pdf',
|
||||
'https://github.com/fenglinglwb/MAT'
|
||||
)
|
||||
default:
|
||||
return <></>
|
||||
}
|
||||
|
||||
@@ -7,6 +7,7 @@ export enum AIModel {
|
||||
LAMA = 'lama',
|
||||
LDM = 'ldm',
|
||||
ZITS = 'zits',
|
||||
MAT = 'mat',
|
||||
}
|
||||
|
||||
export const fileState = atom<File | undefined>({
|
||||
@@ -80,6 +81,12 @@ const defaultHDSettings: ModelsHDSettings = {
|
||||
hdStrategyCropTrigerSize: 1024,
|
||||
hdStrategyCropMargin: 128,
|
||||
},
|
||||
[AIModel.MAT]: {
|
||||
hdStrategy: HDStrategy.CROP,
|
||||
hdStrategyResizeLimit: 1024,
|
||||
hdStrategyCropTrigerSize: 512,
|
||||
hdStrategyCropMargin: 128,
|
||||
},
|
||||
}
|
||||
|
||||
export const settingStateDefault: Settings = {
|
||||
|
||||
Reference in New Issue
Block a user