add Manga Model

2022-11-18 21:40:12 +08:00
parent d7e2148ce1
commit 08c295a70d
7 changed files with 150 additions and 5 deletions
--- a/lama_cleaner/app/src/store/Atoms.tsx
+++ b/lama_cleaner/app/src/store/Atoms.tsx
@@ -11,6 +11,7 @@ export enum AIModel {
  FCF = 'fcf',
  SD15 = 'sd1.5',
  CV2 = 'cv2',
+  Mange = 'manga',
 }

 export const fileState = atom<File | undefined>({
@@ -223,6 +224,13 @@ const defaultHDSettings: ModelsHDSettings = {
    hdStrategyCropMargin: 128,
    enabled: true,
  },
+  [AIModel.Mange]: {
+    hdStrategy: HDStrategy.CROP,
+    hdStrategyResizeLimit: 1280,
+    hdStrategyCropTrigerSize: 1024,
+    hdStrategyCropMargin: 196,
+    enabled: true,
+  },
  [AIModel.CV2]: {
    hdStrategy: HDStrategy.RESIZE,
    hdStrategyResizeLimit: 1080,