pix2pix to instruct_pix2pix

lama_cleaner/app/src/store/Atoms.tsx

@@ -14,7 +14,7 @@ export enum AIModel {
   CV2 = 'cv2',
   Mange = 'manga',
   PAINT_BY_EXAMPLE = 'paint_by_example',
-  PIX2PIX = 'pix2pix',
+  PIX2PIX = 'instruct_pix2pix',
 }
 
 export const maskState = atom<File | undefined>({

lama_cleaner/const.py

@@ -12,7 +12,7 @@ AVAILABLE_MODELS = [
     "manga",
     "sd2",
     "paint_by_example",
-    "pix2pix",
+    "instruct_pix2pix",
 ]
 
 AVAILABLE_DEVICES = ["cuda", "cpu", "mps"]

lama_cleaner/model/instruct_pix2pix.py

@@ -8,7 +8,7 @@ from lama_cleaner.model.utils import set_seed
 from lama_cleaner.schema import Config
 
 
-class Pix2Pix(DiffusionInpaintModel):
+class InstructPix2Pix(DiffusionInpaintModel):
     pad_mod = 8
     min_size = 512
 

lama_cleaner/model_manager.py

@@ -7,14 +7,14 @@ from lama_cleaner.model.ldm import LDM
 from lama_cleaner.model.manga import Manga
 from lama_cleaner.model.mat import MAT
 from lama_cleaner.model.paint_by_example import PaintByExample
-from lama_cleaner.model.pix2pix import Pix2Pix
+from lama_cleaner.model.instruct_pix2pix import InstructPix2Pix
 from lama_cleaner.model.sd import SD15, SD2
 from lama_cleaner.model.zits import ZITS
 from lama_cleaner.model.opencv2 import OpenCV2
 from lama_cleaner.schema import Config
 
 models = {"lama": LaMa, "ldm": LDM, "zits": ZITS, "mat": MAT, "fcf": FcF, "sd1.5": SD15, "cv2": OpenCV2, "manga": Manga,
-          "sd2": SD2, "paint_by_example": PaintByExample, "pix2pix": Pix2Pix}
+          "sd2": SD2, "paint_by_example": PaintByExample, "instruct_pix2pix": InstructPix2Pix}
 
 
 class ModelManager: