"vscode:/vscode.git/clone" did not exist on "6a2a503e376028fcdc87b85c6e08a3daadc412a4"
Unverified Commit 195ebe5a authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

Mark in painting experimental (#430)

parent 1e98723e
......@@ -11,10 +11,14 @@ from transformers import CLIPFeatureExtractor, CLIPTextModel, CLIPTokenizer
from ...models import AutoencoderKL, UNet2DConditionModel
from ...pipeline_utils import DiffusionPipeline
from ...schedulers import DDIMScheduler, PNDMScheduler
from ...utils import logging
from . import StableDiffusionPipelineOutput
from .safety_checker import StableDiffusionSafetyChecker
logger = logging.get_logger(__name__)
def preprocess_image(image):
w, h = image.size
w, h = map(lambda x: x - x % 32, (w, h)) # resize to integer multiple of 32
......@@ -78,6 +82,7 @@ class StableDiffusionInpaintPipeline(DiffusionPipeline):
):
super().__init__()
scheduler = scheduler.set_format("pt")
logger.info("`StableDiffusionInpaintPipeline` is experimental and will very likely change in the future.")
self.register_modules(
vae=vae,
text_encoder=text_encoder,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment