fix import issue

iopaint/model/instruct_pix2pix.py

@@ -3,12 +3,13 @@ import cv2
 import torch
 from loguru import logger
 
+from iopaint.const import INSTRUCT_PIX2PIX_NAME
 from iopaint.model.base import DiffusionInpaintModel
 from iopaint.schema import InpaintRequest
 
 
 class InstructPix2Pix(DiffusionInpaintModel):
-    name = "timbrooks/instruct-pix2pix"
+    name = INSTRUCT_PIX2PIX_NAME
     pad_mod = 8
     min_size = 512
 

iopaint/model/kandinsky.py

@@ -3,8 +3,8 @@ import cv2
 import numpy as np
 import torch
 
+from iopaint.const import KANDINSKY22_NAME
 from iopaint.model.base import DiffusionInpaintModel
-from iopaint.model.utils import get_scheduler
 from iopaint.schema import InpaintRequest
 
 
@@ -62,4 +62,4 @@ class Kandinsky(DiffusionInpaintModel):
 
 
 class Kandinsky22(Kandinsky):
-    name = "kandinsky-community/kandinsky-2-2-decoder-inpaint"
+    name = KANDINSKY22_NAME

iopaint/model/power_paint/power_paint.py

@@ -9,10 +9,11 @@ from iopaint.model.helper.cpu_text_encoder import CPUTextEncoderWrapper
 from iopaint.model.utils import handle_from_pretrained_exceptions
 from iopaint.schema import InpaintRequest
 from .powerpaint_tokenizer import add_task_to_prompt
+from ...const import POWERPAINT_NAME
 
 
 class PowerPaint(DiffusionInpaintModel):
-    name = "Sanster/PowerPaint-V1-stable-diffusion-inpainting"
+    name = POWERPAINT_NAME
     pad_mod = 8
     min_size = 512
     lcm_lora_id = "latent-consistency/lcm-lora-sdv1-5"