From 0ddc1f105c3d64347b5ee5200aecd0bb5a0e24d1 Mon Sep 17 00:00:00 2001 From: Qing Date: Sat, 28 Jan 2023 21:24:51 +0800 Subject: [PATCH] pix2pix to instruct_pix2pix --- lama_cleaner/app/src/store/Atoms.tsx | 2 +- lama_cleaner/const.py | 2 +- lama_cleaner/model/{pix2pix.py => instruct_pix2pix.py} | 2 +- lama_cleaner/model_manager.py | 4 ++-- 4 files changed, 5 insertions(+), 5 deletions(-) rename lama_cleaner/model/{pix2pix.py => instruct_pix2pix.py} (98%) diff --git a/lama_cleaner/app/src/store/Atoms.tsx b/lama_cleaner/app/src/store/Atoms.tsx index c2b51d7..703a828 100644 --- a/lama_cleaner/app/src/store/Atoms.tsx +++ b/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({ diff --git a/lama_cleaner/const.py b/lama_cleaner/const.py index 48ca4c2..e0444e9 100644 --- a/lama_cleaner/const.py +++ b/lama_cleaner/const.py @@ -12,7 +12,7 @@ AVAILABLE_MODELS = [ "manga", "sd2", "paint_by_example", - "pix2pix", + "instruct_pix2pix", ] AVAILABLE_DEVICES = ["cuda", "cpu", "mps"] diff --git a/lama_cleaner/model/pix2pix.py b/lama_cleaner/model/instruct_pix2pix.py similarity index 98% rename from lama_cleaner/model/pix2pix.py rename to lama_cleaner/model/instruct_pix2pix.py index ddcd5a4..1b33e6b 100644 --- a/lama_cleaner/model/pix2pix.py +++ b/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 diff --git a/lama_cleaner/model_manager.py b/lama_cleaner/model_manager.py index 554a038..9ecdc12 100644 --- a/lama_cleaner/model_manager.py +++ b/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: