File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -1274,11 +1274,11 @@ mask_url = "https://raw.githubusercontent.com/CompVis/latent-diffusion/main/data
12741274init_image = download_image(img_url).resize((512 , 512 ))
12751275mask_image = download_image(mask_url).resize((512 , 512 ))
12761276mask_image = PIL .ImageOps.invert(mask_image)
1277- pipe = DiffusionPipeline .from_pretrained(
1277+ pipe = StableDiffusionPipeline .from_pretrained(
12781278 " CompVis/stable-diffusion-v1-4" , torch_dtype = torch.float16, custom_pipeline = " stable_diffusion_repaint" ,
12791279)
12801280pipe.scheduler = RePaintScheduler.from_config(pipe.scheduler.config)
12811281pipe = pipe.to(" cuda" )
12821282prompt = " Face of a yellow cat, high resolution, sitting on a park bench"
12831283image = pipe(prompt = prompt, image = init_image, mask_image = mask_image).images[0 ]
1284- ```
1284+ ```
You can’t perform that action at this time.
0 commit comments