pix2pix to instruct_pix2pix

2023-01-28 21:24:51 +08:00
parent a947f864b6
commit 0ddc1f105c
4 changed files with 5 additions and 5 deletions
--- a/lama_cleaner/model/instruct_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