add pix2pix

lama_cleaner/app/src/store/Atoms.tsx

@@ -14,6 +14,7 @@ export enum AIModel {
   CV2 = 'cv2',
   Mange = 'manga',
   PAINT_BY_EXAMPLE = 'paint_by_example',
+  PIX2PIX = 'pix2pix',
 }
 
 export const maskState = atom<File | undefined>({
@@ -343,6 +344,11 @@ export interface Settings {
   paintByExampleSeedFixed: boolean
   paintByExampleMaskBlur: number
   paintByExampleMatchHistograms: boolean
+
+  // InstructPix2Pix
+  p2pSteps: number
+  p2pImageGuidanceScale: number
+  p2pGuidanceScale: number
 }
 
 const defaultHDSettings: ModelsHDSettings = {
@@ -402,6 +408,13 @@ const defaultHDSettings: ModelsHDSettings = {
     hdStrategyCropMargin: 128,
     enabled: false,
   },
+  [AIModel.PIX2PIX]: {
+    hdStrategy: HDStrategy.ORIGINAL,
+    hdStrategyResizeLimit: 768,
+    hdStrategyCropTrigerSize: 512,
+    hdStrategyCropMargin: 128,
+    enabled: false,
+  },
   [AIModel.Mange]: {
     hdStrategy: HDStrategy.CROP,
     hdStrategyResizeLimit: 1280,
@@ -471,6 +484,11 @@ export const settingStateDefault: Settings = {
   paintByExampleMaskBlur: 5,
   paintByExampleSeedFixed: false,
   paintByExampleMatchHistograms: false,
+
+  // InstructPix2Pix
+  p2pSteps: 50,
+  p2pImageGuidanceScale: 1.5,
+  p2pGuidanceScale: 7.5,
 }
 
 const localStorageEffect =
@@ -567,12 +585,33 @@ export const isPaintByExampleState = selector({
   },
 })
 
+export const isPix2PixState = selector({
+  key: 'isPix2PixState',
+  get: ({ get }) => {
+    const settings = get(settingState)
+    return settings.model === AIModel.PIX2PIX
+  },
+})
+
 export const runManuallyState = selector({
   key: 'runManuallyState',
   get: ({ get }) => {
     const settings = get(settingState)
     const isSD = get(isSDState)
     const isPaintByExample = get(isPaintByExampleState)
-    return settings.runInpaintingManually || isSD || isPaintByExample
+    const isPix2Pix = get(isPix2PixState)
+    return (
+      settings.runInpaintingManually || isSD || isPaintByExample || isPix2Pix
+    )
+  },
+})
+
+export const isDiffusionModelsState = selector({
+  key: 'isDiffusionModelsState',
+  get: ({ get }) => {
+    const isSD = get(isSDState)
+    const isPaintByExample = get(isPaintByExampleState)
+    const isPix2Pix = get(isPix2PixState)
+    return isSD || isPaintByExample || isPix2Pix
   },
 })