kamp/IOPaint
1
0
Fork 0
Code Issues Pull Requests Releases 1 Activity

pix2pix to instruct_pix2pix

Browse Source
This commit is contained in:
Qing 2023-01-28 21:24:51 +08:00
parent a947f864b6
commit 0ddc1f105c
4 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lama_cleaner/app/src/store/Atoms.tsx
View File

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

2
lama_cleaner/const.py
View File

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

2
lama_cleaner/model/pix2pix.py → lama_cleaner/model/instruct_pix2pix.py
View File

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

4
lama_cleaner/model_manager.py
View File

@ -7,14 +7,14 @@ from lama_cleaner.model.ldm import LDM
from lama_cleaner.model.manga import Manga from lama_cleaner.model.manga import Manga
from lama_cleaner.model.mat import MAT from lama_cleaner.model.mat import MAT
from lama_cleaner.model.paint_by_example import PaintByExample 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.sd import SD15, SD2
from lama_cleaner.model.zits import ZITS from lama_cleaner.model.zits import ZITS
from lama_cleaner.model.opencv2 import OpenCV2 from lama_cleaner.model.opencv2 import OpenCV2
from lama_cleaner.schema import Config from lama_cleaner.schema import Config
models = {"lama": LaMa, "ldm": LDM, "zits": ZITS, "mat": MAT, "fcf": FcF, "sd1.5": SD15, "cv2": OpenCV2, "manga": Manga, 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: class ModelManager: