fix import issue

This commit is contained in:
Qing
2024-01-05 16:38:55 +08:00
parent 87d9078fe3
commit f88f3cbbb3
7 changed files with 23 additions and 15 deletions

View File

@@ -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