TensorRT Inpaint pipeline: minor fixes (#4457)

Signed-off-by: Asfiya Baig <asfiyab@nvidia.com>

Author: asfiyab-nvidia

examples/community/README.md

@@ -1646,13 +1646,13 @@ from io import BytesIO
 from PIL import Image
 import torch
 from diffusers import PNDMScheduler
-from diffusers.pipelines.stable_diffusion import StableDiffusionImg2ImgPipeline
+from diffusers.pipelines.stable_diffusion import StableDiffusionInpaintPipeline
 
 # Use the PNDMScheduler scheduler here instead
 scheduler = PNDMScheduler.from_pretrained("stabilityai/stable-diffusion-2-inpainting", subfolder="scheduler")
 
 
-pipe = StableDiffusionImg2ImgPipeline.from_pretrained("stabilityai/stable-diffusion-2-inpainting",
+pipe = StableDiffusionInpaintPipeline.from_pretrained("stabilityai/stable-diffusion-2-inpainting",
     custom_pipeline="stable_diffusion_tensorrt_inpaint",
     revision='fp16',
     torch_dtype=torch.float16,

examples/community/stable_diffusion_tensorrt_inpaint.py

@@ -1054,7 +1054,7 @@ def __call__(
                     return_image=True,
                 ),
             )
-            # print(mask)
+
             mask = torch.nn.functional.interpolate(mask, size=(latent_height, latent_width))
             mask = torch.cat([mask] * 2)