Skip to content

Commit 195ebe5

Browse files
Mark in painting experimental (#430)
1 parent 1e98723 commit 195ebe5

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,10 +11,14 @@
1111
from ...models import AutoencoderKL, UNet2DConditionModel
1212
from ...pipeline_utils import DiffusionPipeline
1313
from ...schedulers import DDIMScheduler, PNDMScheduler
14+
from ...utils import logging
1415
from . import StableDiffusionPipelineOutput
1516
from .safety_checker import StableDiffusionSafetyChecker
1617

1718

19+
logger = logging.get_logger(__name__)
20+
21+
1822
def preprocess_image(image):
1923
w, h = image.size
2024
w, h = map(lambda x: x - x % 32, (w, h)) # resize to integer multiple of 32
@@ -78,6 +82,7 @@ def __init__(
7882
):
7983
super().__init__()
8084
scheduler = scheduler.set_format("pt")
85+
logger.info("`StableDiffusionInpaintPipeline` is experimental and will very likely change in the future.")
8186
self.register_modules(
8287
vae=vae,
8388
text_encoder=text_encoder,

0 commit comments

Comments
 (0)