kandinsky

lama_cleaner/app/src/components/Settings/ModelSettingBlock.tsx

@@ -270,6 +270,12 @@ function ModelSettingBlock() {
           'https://arxiv.org/abs/2211.09800',
           'https://github.com/timothybrooks/instruct-pix2pix'
         )
+      case AIModel.KANDINSKY21:
+        return renderModelDesc(
+          'Kandinsky 2.1',
+          'https://huggingface.co/kandinsky-community/kandinsky-2-1-inpaint',
+          'https://huggingface.co/kandinsky-community/kandinsky-2-1-inpaint'
+        )
       default:
         return <></>
     }

lama_cleaner/app/src/store/Atoms.tsx

@@ -17,6 +17,7 @@ export enum AIModel {
   Mange = 'manga',
   PAINT_BY_EXAMPLE = 'paint_by_example',
   PIX2PIX = 'instruct_pix2pix',
+  KANDINSKY21 = 'kandinsky2.1',
 }
 
 export enum ControlNetMethod {
@@ -566,6 +567,13 @@ const defaultHDSettings: ModelsHDSettings = {
     hdStrategyCropMargin: 128,
     enabled: true,
   },
+  [AIModel.KANDINSKY21]: {
+    hdStrategy: HDStrategy.ORIGINAL,
+    hdStrategyResizeLimit: 768,
+    hdStrategyCropTrigerSize: 512,
+    hdStrategyCropMargin: 128,
+    enabled: false,
+  },
 }
 
 export enum SDSampler {
@@ -719,7 +727,8 @@ export const isSDState = selector({
       settings.model === AIModel.SD15 ||
       settings.model === AIModel.SD2 ||
       settings.model === AIModel.ANYTHING4 ||
-      settings.model === AIModel.REALISTIC_VISION_1_4
+      settings.model === AIModel.REALISTIC_VISION_1_4 ||
+      settings.model === AIModel.KANDINSKY21
     )
   },
 })