Views
No views yet
1import torch
2from diffusers import AutoPipelineForInpainting
3from diffusers.utils import load_image, make_image_grid
4
5pipeline = AutoPipelineForInpainting.from_pretrained(
6 "chuckma/fooocus_inpainting_v2", torch_dtype=torch.float16, variant="fp16"
7)
8pipeline.enable_model_cpu_offload()
9# remove following line if xFormers is not installed or you have PyTorch 2.0 or higher installed
10# pipeline.enable_xformers_memory_efficient_attention()
11
12# load base and mask image
13init_image = load_image("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/inpaint.png")
14mask_image = load_image("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/inpaint_mask.png")
15
16generator = torch.Generator("cuda").manual_seed(92)
17prompt = "concept art digital painting of an elven castle, inspired by lord of the rings, highly detailed, 8k"
18image = pipeline(prompt=prompt, image=init_image, mask_image=mask_image, generator=generator).images[0]
19make_image_grid([init_image, mask_image, image], rows=1, cols=3)