frontend: add ZITS config

2022-07-13 09:04:28 +08:00
parent 1f8fb29594
commit eee91e7b82
4 changed files with 43 additions and 2 deletions
--- a/lama_cleaner/app/src/store/Atoms.tsx
+++ b/lama_cleaner/app/src/store/Atoms.tsx
@@ -46,6 +46,9 @@ export interface Settings {
  // For LDM
  ldmSteps: number
  ldmSampler: LDMSampler
+
+  // For ZITS
+  zitsWireframe: boolean
 }

 export const settingStateDefault = {
@@ -54,8 +57,12 @@ export const settingStateDefault = {
  graduallyInpainting: true,
  runInpaintingManually: false,
  model: AIModel.LAMA,
+
  ldmSteps: 50,
  ldmSampler: LDMSampler.plms,
+
+  zitsWireframe: true,
+
  hdStrategy: HDStrategy.RESIZE,
  hdStrategyResizeLimit: 2048,
  hdStrategyCropTrigerSize: 2048,