correct pipeline class name (#5652)

aec3de8b · Sayak Paul · GitHub · d61889fc · aec3de8b
Unverified Commit aec3de8b authored Nov 06, 2023 by Sayak Paul Committed by GitHub Nov 06, 2023
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 3 deletions

src/diffusers/pipelines/pixart_alpha/pipeline_pixart_alpha.py ...diffusers/pipelines/pixart_alpha/pipeline_pixart_alpha.py +1 -3

No files found.
--- a/src/diffusers/pipelines/pixart_alpha/pipeline_pixart_alpha.py
+++ b/src/diffusers/pipelines/pixart_alpha/pipeline_pixart_alpha.py
@@ -51,9 +51,7 @@ EXAMPLE_DOC_STRING = """
        >>> from diffusers import PixArtAlphaPipeline

        >>> # You can replace the checkpoint id with "PixArt-alpha/PixArt-XL-2-512x512" too.
-        >>> pipe = StableDiffusionXLPipeline.from_pretrained(
-        ...     "PixArt-alpha/PixArt-XL-2-1024-MS", torch_dtype=torch.float16
-        ... )
+        >>> pipe = PixArtAlphaPipeline.from_pretrained("PixArt-alpha/PixArt-XL-2-1024-MS", torch_dtype=torch.float16)
        >>> # Enable memory optimizations.
        >>> pipe.enable_model_cpu_offload()